bluenevus commited on
Commit
261d49a
·
verified ·
1 Parent(s): 29f0e9b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
20
  print(f"Using device: {device}")
21
 
22
  # Load the Whisper model and processor
23
- whisper_model_name = "openai/whisper-small.en"
24
  whisper_processor = WhisperProcessor.from_pretrained(whisper_model_name)
25
  whisper_model = WhisperForConditionalGeneration.from_pretrained(whisper_model_name).to(device)
26
 
 
20
  print(f"Using device: {device}")
21
 
22
  # Load the Whisper model and processor
23
+ whisper_model_name = "openai/whisper-small"
24
  whisper_processor = WhisperProcessor.from_pretrained(whisper_model_name)
25
  whisper_model = WhisperForConditionalGeneration.from_pretrained(whisper_model_name).to(device)
26