M2SA-demo-multimodal

Runtime error

thak123 commited on May 20

Commit

6fe0480

•

1 Parent(s): 169cd2b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -160,8 +160,8 @@ def predict_sentiment(text, image):
         outputs = model(**text_inputs)
         print(outputs)
         prediction = np.argmax(outputs["logits"], axis=-1)
-        print(id2label[prediction[0]])
-    return prediction
 interface = gr.Interface(

         outputs = model(**text_inputs)
         print(outputs)
         prediction = np.argmax(outputs["logits"], axis=-1)
+        print(id2label[prediction[0].item()])
+    return id2label[prediction[0].item()]
 interface = gr.Interface(