Hugging Face's logo

Spaces:
nunenuh
/
whisper_simple
Runtime error

App Files Community
nunenuh commited on
Commit
8d023c9
•
1 Parent(s): 4b068e8

feat: change model to whishper medium

Browse files
Files changed (1) hide show
  1. src/infer.py +1 -1
src/infer.py CHANGED
@@ -3,7 +3,7 @@ from typing import *
3
  from src import utils
4
  from transformers import WhisperProcessor, WhisperForConditionalGeneration
5
 
6
- model_name: str = f"openai/whisper-small"
7
  processor: Any = WhisperProcessor.from_pretrained(model_name)
8
  model: Any = WhisperForConditionalGeneration.from_pretrained(model_name)
9
 
 
3
  from src import utils
4
  from transformers import WhisperProcessor, WhisperForConditionalGeneration
5
 
6
+ model_name: str = f"openai/whisper-medium"
7
  processor: Any = WhisperProcessor.from_pretrained(model_name)
8
  model: Any = WhisperForConditionalGeneration.from_pretrained(model_name)
9