Update app.py

app.py

@@ -4,8 +4,11 @@ from transformers import Wav2Vec2ForCTC, Wav2Vec2Processor
 import gradio as gr
 import sox
 import subprocess
+from google_spell_checker import GoogleSpellChecker
 
 
+spell_checker = GoogleSpellChecker(lang="fa")
+
 def read_file_and_process(wav_file):
     filename = wav_file.split('.')[0]
     filename_16k = filename + "16k.wav"
@@ -34,7 +37,8 @@ def parse(wav_file):
     input_values = read_file_and_process(wav_file)
     with torch.no_grad():
         logits = model(**input_values).logits
-    return parse_transcription(logits)
+        sentence = parse_transcription(logits)
+    return spell_checker.check(sentence)[1]
 
 model_id = "jonatasgrosman/wav2vec2-large-xlsr-53-persian"
 processor = Wav2Vec2Processor.from_pretrained(model_id)