bstraehle commited on
Commit
5b45741
1 Parent(s): b4de4c9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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 == 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_2:
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