aiola commited on
Commit
6eff469
·
verified ·
1 Parent(s): 94d76bf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ import spaces
8
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
9
 
10
  # Load model and processor
11
- processor = WhisperProcessor.from_pretrained("openai/whisper-large-v2")
12
  model = WhisperForConditionalGeneration.from_pretrained("aiola/whisper-ner-v1")
13
  model = model.to(device)
14
 
 
8
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
9
 
10
  # Load model and processor
11
+ processor = WhisperProcessor.from_pretrained("aiola/whisper-ner-v1")
12
  model = WhisperForConditionalGeneration.from_pretrained("aiola/whisper-ner-v1")
13
  model = model.to(device)
14