eleferrand commited on
Commit
91eb016
·
verified ·
1 Parent(s): b9cdbc0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -39,6 +39,13 @@ def transcribe(audio_file):
39
  transc = transcription.replace("[UNK]", "")
40
  print(transc)
41
  output.append((start,ind,transc))
 
 
 
 
 
 
 
42
  return output
43
 
44
 
 
39
  transc = transcription.replace("[UNK]", "")
40
  print(transc)
41
  output.append((start,ind,transc))
42
+ else:
43
+ input_values = processor(audio, sampling_rate=16000, return_tensors="pt").input_values
44
+ with torch.no_grad():
45
+ logits = model(input_values).logits
46
+ predicted_ids = torch.argmax(logits, dim=-1)
47
+ transcription = processor.batch_decode(predicted_ids)[0]
48
+ output.append((start,len(audio)/sr),transcription)
49
  return output
50
 
51