thak123 commited on
Commit
2e53777
1 Parent(s): 05e22ed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -16,6 +16,7 @@ from safetensors.torch import load_model
16
  from datasets import load_dataset, load_metric
17
  from transformers import (
18
  AutoConfig,
 
19
  AutoModelForSequenceClassification,
20
  AutoTokenizer,
21
  logging,
@@ -144,7 +145,7 @@ def predict_sentiment(text, image):
144
  }
145
  prediction = None
146
  with torch.no_grad():
147
- prediction = model(*model_input)
148
  print(prediction)
149
  return prediction
150
 
 
16
  from datasets import load_dataset, load_metric
17
  from transformers import (
18
  AutoConfig,
19
+ AutoImageProcessor,
20
  AutoModelForSequenceClassification,
21
  AutoTokenizer,
22
  logging,
 
145
  }
146
  prediction = None
147
  with torch.no_grad():
148
+ prediction = model(**model_input)
149
  print(prediction)
150
  return prediction
151