Update app.py

app.py

@@ -94,11 +94,11 @@ load_model(model, sf_filename)
 # model.load_state_dict(torch.load(model_args.model_name_or_path+"-finetuned/pytorch_model.bin"))
 
 
-model = AutoModelForSequenceClassification.from_pretrained(
-        "FFZG-cleopatra/M2SA",
-        num_labels=3, id2label=id2label,
-        label2id=label2id
-    )
+# model = AutoModelForSequenceClassification.from_pretrained(
+#         "FFZG-cleopatra/M2SA",
+#         num_labels=3, id2label=id2label,
+#         label2id=label2id
+#     )
 
 
 def predict_sentiment(text, image):
@@ -124,7 +124,7 @@ def predict_sentiment(text, image):
     }
     prediction = None
     with torch.no_grad():
-        prediction = model(model_input)
+        prediction = model(*model_input)
         print(prediction)
     return prediction