Update app.py

app.py

@@ -149,19 +149,15 @@ def predict_sentiment(text, image):
         image_processor.image_std,
     )
     image_transformations = torch.jit.script(image_transformations)
-    image = image_transformations(image)
+    pixel_values = image_transformations(image)
     
-    model_input = {
-        "input_ids" : text_inputs.input_ids,
-        "pixel_values":image,
-        "attention_mask" : text_inputs.attention_mask,
-    }
+   
     print(text_inputs)
-    print("its going in ",image)
-    print(model_input)
+    print("its going in ",pixel_values)
+
     prediction = None
     with torch.no_grad():
-        prediction = model(input_ids=[text_inputs.input_id],attention_mask=[text_inputs.attention_mask], pixel_values=[image])
+        prediction = model(input_ids=text_inputs.input_id,attention_mask=text_inputs.attention_mask, pixel_values=pixel_values)
         print(prediction)
     return prediction