Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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(
|
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 |
|