Update app.py
Browse files
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 =
|
91 |
-
gr.Textbox(label = "User Prompt", value = user_prompt, lines =
|
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)],
|