kshubham2107 commited on
Commit
2fc74a9
·
verified ·
1 Parent(s): 7cf86ef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,7 +9,7 @@ import gradio as gr
9
  openai.api_key = os.getenv("OPENAI_API_KEY")
10
  fine_tuned_model_id = os.getenv("fine_tuned_model_id")
11
 
12
- def fine_tune_test(prompt, system_message="Act as a perfectionist and map the given product by doing critical reasoning."):
13
  test_messages = []
14
  test_messages.append({"role": "system", "content": system_message})
15
  user_message = prompt
@@ -23,7 +23,7 @@ def fine_tune_test(prompt, system_message="Act as a perfectionist and map the gi
23
  print(response["choices"][0]["message"]["content"])
24
  return(response["choices"][0]["message"]["content"])
25
 
26
- iface = gr.Interface(fn=fine_tune_test,inputs=[gr.Textbox(label="User Prompt"),gr.Textbox(label="System Prompt")],
27
  outputs=[gr.Textbox(label="Fine Tuned GPT Repsonse")],
28
  title="GPT 3.5-Turbo Fine Tuning")
29
 
 
9
  openai.api_key = os.getenv("OPENAI_API_KEY")
10
  fine_tuned_model_id = os.getenv("fine_tuned_model_id")
11
 
12
+ def fine_tune_test(system_message, prompt):
13
  test_messages = []
14
  test_messages.append({"role": "system", "content": system_message})
15
  user_message = prompt
 
23
  print(response["choices"][0]["message"]["content"])
24
  return(response["choices"][0]["message"]["content"])
25
 
26
+ iface = gr.Interface(fn=fine_tune_test,inputs=[gr.Textbox(label="System Prompt", value="Act as a perfectionist and map the given product by doing critical reasoning."), gr.Textbox(label="User Prompt")],
27
  outputs=[gr.Textbox(label="Fine Tuned GPT Repsonse")],
28
  title="GPT 3.5-Turbo Fine Tuning")
29