Update app.py
Browse files
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([
|
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),
|