Update app.py

app.py

@@ -3,8 +3,7 @@ import wikipedia
 os.system("pip install git+https://github.com/openai/whisper.git")
 import gradio as gr
 import whisper
-from jiwer import wer
-
+import jiwer
 from share_btn import community_icon_html, loading_icon_html, share_js
 
 model = whisper.load_model("small")
@@ -24,12 +23,25 @@ def inference(audio):
     options = whisper.DecodingOptions(fp16 = False)
     result = whisper.decode(model, mel, options)
     
-    error = wer(passage, result.text)
-    print(f"WER is {error}")
+    transformation = jiwer.Compose([
+        jiwer.ToLowerCase(),
+        jiwer.RemoveWhiteSpace(replace_by_space=True),
+        jiwer.RemoveMultipleSpaces(),
+        jiwer.ReduceToListOfListOfWords(word_delimiter=" ")
+    ]) 
     
+    error = jiwer.wer(
+        passage, 
+        result.text, 
+        truth_transform=transformation, 
+        hypothesis_transform=transformation
+       )
+    # error = jiwer.wer(passage, result.text)
+    print(f"WER is {error * len(result.text.split())}")
     print(result.text)
     print(passage)
-    return result.text, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
+    
+    return f"WER is {error * len(result.text.split())} \n\n{result.text}", gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)