Spaces:
Running
on
T4
Running
on
T4
Update app.py
Browse files
app.py
CHANGED
|
@@ -86,7 +86,7 @@ def get_audio_from_youtube(url):
|
|
| 86 |
|
| 87 |
|
| 88 |
def get_transcripts(audio_path, model):
|
| 89 |
-
with torch.cuda.amp.autocast(enabled=
|
| 90 |
with torch.inference_mode():
|
| 91 |
text = model.transcribe(audio=[audio_path], )
|
| 92 |
return text
|
|
|
|
| 86 |
|
| 87 |
|
| 88 |
def get_transcripts(audio_path, model):
|
| 89 |
+
with torch.cuda.amp.autocast(enabled=False, dtype=torch.bfloat16):
|
| 90 |
with torch.inference_mode():
|
| 91 |
text = model.transcribe(audio=[audio_path], )
|
| 92 |
return text
|