romsyflux commited on
Commit
daf6cbd
1 Parent(s): 3a3a6e5

Update Whisper model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = "distil-whisper/distil-small.en"
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
  )