Update app.py

app.py

@@ -24,10 +24,10 @@ def load_data(input_file):
     speech = librosa.resample(speech, sample_rate,16000)
   return speech
 
-def correct_casing(input_sentence):
+# def correct_casing(input_sentence):
 
-  sentences = nltk.sent_tokenize(input_sentence)
-  return (' '.join([s.replace(s[0],s[0].capitalize(),1) for s in sentences]))
+#   sentences = nltk.sent_tokenize(input_sentence)
+#   return (' '.join([s.replace(s[0],s[0].capitalize(),1) for s in sentences]))
 
 def asr_transcript(input_file):
 
@@ -40,9 +40,9 @@ def asr_transcript(input_file):
   #Take argmax
   predicted_ids = torch.argmax(logits, dim=-1)
   #Get the words from predicted word ids
-  transcription = tokenizer.decode(predicted_ids[0])
+      transcription = processor.batch_decode(predicted_ids[0])
   #Correcting the letter casing
-  transcription = correct_casing(transcription.lower())
+  #transcription = correct_casing(transcription.lower())
   return transcription
 
 gr.Interface(asr_transcript,