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
|