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 |
|