thak123 commited on
Commit
45c6dc1
1 Parent(s): 7dd6171

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -150,7 +150,7 @@ def predict_sentiment(text, image):
150
  )
151
  image_transformations = torch.jit.script(image_transformations)
152
  pixel_values = image_transformations(image)
153
- text_inputs["pixel_values"] = pixel_values
154
 
155
  print(text_inputs)
156
  print("its going in ",pixel_values)
 
150
  )
151
  image_transformations = torch.jit.script(image_transformations)
152
  pixel_values = image_transformations(image)
153
+ text_inputs["pixel_values"] = [pixel_values]
154
 
155
  print(text_inputs)
156
  print("its going in ",pixel_values)