Update app.py

app.py

@@ -158,14 +158,16 @@ class CustomEncoderWav2vec2Classifier(Pretrained):
 
 
 def return_prediction(mic, file):
-    # classifier = foreign_class(source="Porjaz/wavlm-base-emo-fi", pymodule_file="custom_interface.py", classname="CustomEncoderWav2vec2Classifier")
     if mic is not None:
         out_prob, score, index, text_lab = classifier.classify_file(mic)
     elif file is not None:
         out_prob, score, index, text_lab = classifier.classify_file(file)
     else:
         return "You must either provide a mic recording or a file"
-    return text_lab
+    
+    score = score.item()
+    score = str(round(100 * score, 2)) + "%"
+    return text_lab, score
     
 
 classifier = foreign_class(source="Porjaz/wavlm-base-emo-fi", pymodule_file="custom_interface.py", classname="CustomEncoderWav2vec2Classifier")
@@ -180,4 +182,6 @@ gradio_app = gr.Interface(
     title="Finnish-Emotion-Recognition",
 )
 
-gradio_app.launch(share=True, ssl_verify=False)
+
+if __name__ == "__main__":
+    gradio_app.launch(share=True, ssl_verify=False)