Fix default parameters

app.py

@@ -8,22 +8,22 @@ import torch
 
 directory = 'av-generation/'
 
-def predict(title, type, base_model, dataset, num_beams):
-    if type == "End2End":
+def predict(title, approach='End2End', base_model='t5-small', dataset='oa-mine', num_beams=3):
+    if approach == "End2End":
         model_id = directory + f"{base_model.lower()}-{type.lower()}-{dataset.lower()}"
         model = TransformersAVG(model_id)
         predictions = model.generate_av_end2end(title, num_beams=num_beams)   
-    elif type == "Pipeline":
+    elif approach == "Pipeline":
         ag_model_id = directory + f"{base_model.lower()}-ag-{dataset.lower()}"
         ve_model_id = directory + f"{base_model.lower()}-ve-{dataset.lower()}"
         model = TransformersAVG(ag_model_id, model_ve=ve_model_id)
         predictions = model.generate_av_pipeline(title, num_beams=num_beams)
-    elif type == 'Multitask':
+    elif approach == 'Multitask':
         model_id = directory + f"{base_model.lower()}-mlt-{dataset.lower()}"
         model = TransformersAVG(model_id)
         predictions = model.generate_av_mul(title, num_beams=num_beams)
     else:
-        pass
+         gr. Error("Please select AVG approach!")
     df = pd.DataFrame(predictions, columns=['Attribute', 'Value'])
     return gr.Dataframe(df)