chinmaydan commited on
Commit
d37215e
·
1 Parent(s): da91c46

changed froom large model to medium

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,8 +4,8 @@ import torch
4
 
5
  from transformers import WhisperProcessor, WhisperForConditionalGeneration
6
 
7
- processor = WhisperProcessor.from_pretrained("openai/whisper-large")
8
- model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-large")
9
 
10
  model.config.forced_decoder_ids = None
11
 
 
4
 
5
  from transformers import WhisperProcessor, WhisperForConditionalGeneration
6
 
7
+ processor = WhisperProcessor.from_pretrained("openai/whisper-medium")
8
+ model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-medium")
9
 
10
  model.config.forced_decoder_ids = None
11