Samfredoly commited on
Commit
2960005
·
verified ·
1 Parent(s): 8f8df59

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -44,7 +44,7 @@ os.makedirs(LOCAL_STATE_FOLDER, exist_ok=True)
44
  # Whisper Model Setup (using transformers)
45
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
46
  TORCH_DTYPE = torch.float16 if torch.cuda.is_available() else torch.float32
47
- WHISPER_MODEL_ID = f"openai/whisper-{WHISPER_MODEL}"
48
 
49
  # Global model cache
50
  _whisper_model = None
 
44
  # Whisper Model Setup (using transformers)
45
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
46
  TORCH_DTYPE = torch.float16 if torch.cuda.is_available() else torch.float32
47
+ WHISPER_MODEL_ID = f"openai/whisper-small"
48
 
49
  # Global model cache
50
  _whisper_model = None