bstraehle commited on
Commit
94ca6da
1 Parent(s): df9a90e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -21,7 +21,7 @@ action_1 = "Prompt base model"
21
  action_2 = "Prompt fine-tuned model"
22
  action_3 = "Fine-tune base model"
23
 
24
- system_prompt = "You are a text to SQL query translator. A user will ask you a question in English and you will generate a SQL query. Do not generate any additional text."
25
  user_prompt = "What is the total trade value and average price for each trader and stock in the trade_history table?"
26
 
27
  base_model_id = "meta-llama/Meta-Llama-3-8B-Instruct"
@@ -87,8 +87,8 @@ def process(action, system_prompt, user_prompt, base_model_id, fine_tuned_model_
87
 
88
  demo = gr.Interface(fn=process,
89
  inputs=[gr.Radio([action_1, action_2, action_3], label = "Action", value = action_2),
90
- gr.Textbox(label = "System Prompt", value = system_prompt, lines = 1),
91
- gr.Textbox(label = "User Prompt", value = user_prompt, lines = 1),
92
  gr.Textbox(label = "Base Model ID", value = base_model_id, lines = 1),
93
  gr.Textbox(label = "Fine-Tuned Model ID", value = fine_tuned_model_id, lines = 1),
94
  gr.Textbox(label = "Dataset", value = dataset, lines = 1)],
 
21
  action_2 = "Prompt fine-tuned model"
22
  action_3 = "Fine-tune base model"
23
 
24
+ system_prompt = "You are a text to SQL query translator. A user will ask you a question in English and you will generate a SQL query. Do will not generate any additional text."
25
  user_prompt = "What is the total trade value and average price for each trader and stock in the trade_history table?"
26
 
27
  base_model_id = "meta-llama/Meta-Llama-3-8B-Instruct"
 
87
 
88
  demo = gr.Interface(fn=process,
89
  inputs=[gr.Radio([action_1, action_2, action_3], label = "Action", value = action_2),
90
+ gr.Textbox(label = "System Prompt", value = system_prompt, lines = 2),
91
+ gr.Textbox(label = "User Prompt", value = user_prompt, lines = 2),
92
  gr.Textbox(label = "Base Model ID", value = base_model_id, lines = 1),
93
  gr.Textbox(label = "Fine-Tuned Model ID", value = fine_tuned_model_id, lines = 1),
94
  gr.Textbox(label = "Dataset", value = dataset, lines = 1)],