thak123 commited on
Commit
946cc43
1 Parent(s): d8b6d5c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -150,14 +150,14 @@ def predict_sentiment(text, image):
150
  )
151
  image_transformations = torch.jit.script(image_transformations)
152
  pixel_values = image_transformations(image)
153
-
154
 
155
  print(text_inputs)
156
  print("its going in ",pixel_values)
157
 
158
  prediction = None
159
  with torch.no_grad():
160
- prediction = model(*text_inputs, pixel_values=pixel_values)
161
  print(prediction)
162
  return prediction
163
 
 
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)
157
 
158
  prediction = None
159
  with torch.no_grad():
160
+ prediction = model(*text_inputs)
161
  print(prediction)
162
  return prediction
163