Update app.py
Browse files
app.py
CHANGED
@@ -18,11 +18,11 @@ DATASET_NAME = "gretelai/synthetic_text_to_sql"
|
|
18 |
|
19 |
def process(action, base_model_name, ft_model_name, dataset_name, system_prompt, user_prompt, sql_schema):
|
20 |
#raise gr.Error("Please clone and bring your own credentials.")
|
21 |
-
if action ==
|
22 |
result = prompt_model(base_model_name, system_prompt, user_prompt, sql_schema)
|
23 |
-
elif action ==
|
24 |
result = fine_tune_model(base_model_name, dataset_name)
|
25 |
-
elif action ==
|
26 |
result = prompt_model(ft_model_name, system_prompt, user_prompt, sql_schema)
|
27 |
return result
|
28 |
|
|
|
18 |
|
19 |
def process(action, base_model_name, ft_model_name, dataset_name, system_prompt, user_prompt, sql_schema):
|
20 |
#raise gr.Error("Please clone and bring your own credentials.")
|
21 |
+
if action == ACTION_1:
|
22 |
result = prompt_model(base_model_name, system_prompt, user_prompt, sql_schema)
|
23 |
+
elif action == ACTION_2:
|
24 |
result = fine_tune_model(base_model_name, dataset_name)
|
25 |
+
elif action == ACTION_3:
|
26 |
result = prompt_model(ft_model_name, system_prompt, user_prompt, sql_schema)
|
27 |
return result
|
28 |
|