Update app.py

app.py

@@ -21,13 +21,15 @@ def read_file(wav):
     return resampled_signal
 
 
+
+
 def parse_transcription_with_lm(wav_file):
     input_values = read_file_and_process(wav_file)
 
     with torch.no_grad():
-        logits = model(**input_values).logits[0]
+        logits = model(**input_values).logits.cpu().numpy()
     print(logits)
-    int_result = processor_with_LM.decode(logits)
+    int_result = processor_with_LM.decode(logits = logits, output_word_offsets=False)
     print(int_result)
     transcription =  int_result
     return transcription