thak123 commited on
Commit
ffc2eef
·
verified ·
1 Parent(s): 935135a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -154,7 +154,7 @@ def predict_sentiment(text, image):
154
  }
155
  prediction = None
156
  with torch.no_grad():
157
- prediction = model(**model_input)
158
  print(prediction)
159
  return prediction
160
 
 
154
  }
155
  prediction = None
156
  with torch.no_grad():
157
+ prediction = model(input_ids=text_inputs.input_ids,attention_mask=text_inputs.attention_mask, pixel_values=image)
158
  print(prediction)
159
  return prediction
160