Lum4yx commited on
Commit
d1e4dc7
·
verified ·
1 Parent(s): 2eebca5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ device = "cuda:0" if torch.cuda.is_available() else "cpu"
11
  torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
12
 
13
  # 2. Define the model ID for the large Whisper model
14
- model_id = "openai/whisper-large-v3-turbo"
15
 
16
  # 3. Load the model from pretrained weights
17
  model = AutoModelForSpeechSeq2Seq.from_pretrained(
 
11
  torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
12
 
13
  # 2. Define the model ID for the large Whisper model
14
+ model_id = "openai/whisper-small"
15
 
16
  # 3. Load the model from pretrained weights
17
  model = AutoModelForSpeechSeq2Seq.from_pretrained(