Spaces:
Runtime error
Runtime error
romsyflux
commited on
Commit
•
daf6cbd
1
Parent(s):
3a3a6e5
Update Whisper model
Browse files
app.py
CHANGED
@@ -15,7 +15,7 @@ print(f"Using device: {device}")
|
|
15 |
torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
|
16 |
|
17 |
# Model and pipeline setup
|
18 |
-
model_id = "
|
19 |
model = AutoModelForSpeechSeq2Seq.from_pretrained(
|
20 |
model_id, torch_dtype=torch_dtype, low_cpu_mem_usage=True, use_safetensors=True
|
21 |
)
|
|
|
15 |
torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
|
16 |
|
17 |
# Model and pipeline setup
|
18 |
+
model_id = "openai/whisper-large-v3"
|
19 |
model = AutoModelForSpeechSeq2Seq.from_pretrained(
|
20 |
model_id, torch_dtype=torch_dtype, low_cpu_mem_usage=True, use_safetensors=True
|
21 |
)
|