Fix dataset name

app.py

@@ -8,7 +8,7 @@ import torch
 
 directory = 'av-generation/'
 
-def predict(title, approach='End2End', base_model='t5-small', dataset='oa-mine', num_beams=3):
+def predict(title, approach='End2End', base_model='t5-small', dataset='AE-110K', num_beams=3):
     if approach == "End2End":
         model_id = directory + f"{base_model.lower()}-{approach.lower()}-{dataset.lower()}"
         model = TransformersAVG(model_id)