thak123 commited on
Commit
d8b6d5c
·
verified ·
1 Parent(s): 13a86d8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -157,7 +157,7 @@ def predict_sentiment(text, image):
157
 
158
  prediction = None
159
  with torch.no_grad():
160
- prediction = model(input_ids=text_inputs.input_id,attention_mask=text_inputs.attention_mask, pixel_values=pixel_values)
161
  print(prediction)
162
  return prediction
163
 
 
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