Spaces:

Porjaz
/

wavlm-base-emo-fi

Sleeping

Porjaz commited on Mar 22

Commit

6256cb3

•

1 Parent(s): ea6c301

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -193,7 +193,7 @@ class CustomEncoderWav2vec2Classifier(Pretrained):
 def return_prediction(mic, file):
-    # classifier = foreign_class(source="Porjaz/wavlm-base-emo-fi", pymodule_file="custom_interface.py", classname="CustomEncoderWav2vec2Classifier")
     out_prob, score, index, text_lab = classifier.classify_file("anger.wav")
     return text_lab

 def return_prediction(mic, file):
+    classifier = foreign_class(source="Porjaz/wavlm-base-emo-fi", pymodule_file="custom_interface.py", classname="CustomEncoderWav2vec2Classifier")
     out_prob, score, index, text_lab = classifier.classify_file("anger.wav")
     return text_lab