Raemih commited on
Commit
ec48379
·
verified ·
1 Parent(s): cee5ba4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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
- ).to(device)
 
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