Hugging Face's logo

Spaces:
sohamchitimali
/
whisper
Running

App Files Community
sohamchitimali commited on
Commit
545ab2e
·
1 Parent(s): b997d93

update

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ from fastapi.middleware.cors import CORSMiddleware
8
  import uvicorn
9
 
10
  # 🔹 Load Whisper model
11
- MODEL_NAME = os.getenv("WHISPER_MODEL", "base")
12
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
13
  MODEL = whisper.load_model(MODEL_NAME, device=DEVICE)
14
 
 
8
  import uvicorn
9
 
10
  # 🔹 Load Whisper model
11
+ MODEL_NAME = os.getenv("WHISPER_MODEL", "small")
12
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
13
  MODEL = whisper.load_model(MODEL_NAME, device=DEVICE)
14