Update default values

app.py

@@ -8,7 +8,7 @@ import torch
 
 directory = 'av-generation/'
 
-def predict(title, type='End2End', base_model='t5-small', dataset='ae-110k', num_beams=3):
+def predict(title, type, base_model, dataset, num_beams):
     if type == "End2End":
         model_id = directory + f"{base_model.lower()}-{type.lower()}-{dataset.lower()}"
         model = TransformersAVG(model_id)
@@ -64,7 +64,7 @@ demo = gr.Interface(
 
         gr.Radio(["AE-110K", "OA-Mine"], value=["AE-110K"], label="Dataset", info="Select dataset."),
 
-        gr.Slider(1, 10, value=4, step=1, label="Number of Beams", info="Degree of exploration at inference.")
+        gr.Slider(1, 10, value=3, step=1, label="Number of Beams", info="Degree of exploration at inference.")
     ],
     "dataframe",
     title="Attribute Value Generation",