Update app.py

app.py

@@ -157,8 +157,10 @@ def predict_sentiment(text, image):
 
     prediction = None
     with torch.no_grad():
-        prediction = model(**text_inputs)
-        print(prediction)
+        outputs = model(**text_inputs)
+        print(outputs)
+        prediction = np.argmax(outputs["logits"], axis=-1)
+        print(id2label[prediction[0]])
     return prediction