Update app.py

app.py

@@ -134,7 +134,7 @@ def predict_sentiment(text, image):
     print(dir(image))
     image = read_image(image, mode=ImageReadMode.RGB)
     # image = transforms.ToTensor()(image).unsqueeze(0)
-    
+    print(image)
     text_inputs = tokenizer(
             text,
             max_length=512,
@@ -150,17 +150,18 @@ def predict_sentiment(text, image):
     )
     image_transformations = torch.jit.script(image_transformations)
     image = image_transformations(image)
+    
     model_input = {
         "input_ids" : text_inputs.input_ids,
         "pixel_values":image,
         "attention_mask" : text_inputs.attention_mask,
     }
     print(text_inputs)
-    print(image)
+    print("its going in ",image)
     print(model_input)
     prediction = None
     with torch.no_grad():
-        prediction = model(input_ids=text_inputs.input_ids,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=[image])
         print(prediction)
     return prediction