Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
|