Update app.py

app.py

@@ -37,8 +37,14 @@ def parse(wav_file):
     input_values = read_file_and_process(wav_file)
     with torch.no_grad():
         logits = model(**input_values).logits
-        sentence = parse_transcription(logits)
-    return spell_checker.check(sentence)[1]
+    sentence = parse_transcription(logits)
+    check_spell = spell_checker.check(sentence)
+    if check_spell[0] is False:
+      sentence = sentence 
+    elif check_spell[0] is True:
+      sentence = check_spell[1]
+    return sentence
+
 
 model_id = "jonatasgrosman/wav2vec2-large-xlsr-53-persian"
 processor = Wav2Vec2Processor.from_pretrained(model_id)