Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -21,8 +21,9 @@ print("Loading model...")
|
|
| 21 |
feature_extractor = AutoFeatureExtractor.from_pretrained(MODEL_ID)
|
| 22 |
|
| 23 |
emotion_model = MMSForMultilingualSER.from_pretrained(
|
| 24 |
-
MODEL_ID
|
| 25 |
-
|
|
|
|
| 26 |
|
| 27 |
emotion_model.eval()
|
| 28 |
|
|
|
|
| 21 |
feature_extractor = AutoFeatureExtractor.from_pretrained(MODEL_ID)
|
| 22 |
|
| 23 |
emotion_model = MMSForMultilingualSER.from_pretrained(
|
| 24 |
+
MODEL_ID,
|
| 25 |
+
ignore_mismatched_sizes=True
|
| 26 |
+
)
|
| 27 |
|
| 28 |
emotion_model.eval()
|
| 29 |
|