Update app.py

app.py

@@ -9,7 +9,7 @@ def return_model(task):
         model = pipeline("text-classification", model="CesarLeblanc/test_model")
     else:
         model = pipeline("fill-mask", model="CesarLeblanc/fill_mask_model")
-    return return_model
+    return model
 
 def return_dataset():
     dataset = load_dataset("CesarLeblanc/text_classification_dataset")