hotfix for greedy search

app.py

@@ -27,7 +27,7 @@ def fix_transcription_casing(input_sentence):
   sentences = nltk.sent_tokenize(input_sentence)
   return (' '.join([s.replace(s[0],s[0].capitalize(),1) for s in sentences]))
   
-def predict_and_decode(input_file):
+def predict_and_ctc_decode(input_file):
   speech = load_and_fix_data(input_file)
 
   input_values = processor(speech, return_tensors="pt", sampling_rate=16000).input_values
@@ -50,12 +50,12 @@ def predict_and_greedy_decode(input_file):
   predicted_ids = torch.argmax(logits, dim=-1)
   pred = processor.batch_decode(predicted_ids)
 
-  transcribed_text = fix_transcription_casing(pred.lower())
+  transcribed_text = fix_transcription_casing(pred[0].lower())
 
   return transcribed_text
 
 def return_all_predictions(input_file):
-  return predict_and_decode(input_file), predict_and_greedy_decode(input_file)
+  return predict_and_ctc_decode(input_file), predict_and_greedy_decode(input_file)
 
 
 gr.Interface(return_all_predictions,