bstraehle commited on
Commit
b4de4c9
·
verified ·
1 Parent(s): 2dfbd8a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -136,7 +136,7 @@ def load_model(model_name):
136
  return model, tokenizer
137
 
138
  demo = gr.Interface(fn=process,
139
- inputs=[gr.Radio([action_1, action_2], label = "Action", value = action_2),
140
  gr.Textbox(label = "Base Model Name", value = BASE_MODEL_NAME, lines = 1),
141
  gr.Textbox(label = "Fine-Tuned Model Name", value = FT_MODEL_NAME, lines = 1),
142
  gr.Textbox(label = "Dataset Name", value = DATASET_NAME, lines = 1),
 
136
  return model, tokenizer
137
 
138
  demo = gr.Interface(fn=process,
139
+ inputs=[gr.Radio([ACTION_1, ACTION_2, ACTION_3], label = "Action", value = ACTION_3),
140
  gr.Textbox(label = "Base Model Name", value = BASE_MODEL_NAME, lines = 1),
141
  gr.Textbox(label = "Fine-Tuned Model Name", value = FT_MODEL_NAME, lines = 1),
142
  gr.Textbox(label = "Dataset Name", value = DATASET_NAME, lines = 1),