legusxyz commited on
Commit
35963ea
·
verified ·
1 Parent(s): 6d35bfe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ app = FastAPI()
11
  # Load the Whisper model once during startup
12
  # device = 0 if torch.cuda.is_available() else -1 # Use GPU if available, otherwise CPU
13
  device = -1
14
- asr_pipeline = pipeline(model="openai/whisper-small", device=device) # Initialize Whisper model
15
 
16
  # Basic GET endpoint
17
  @app.get("/")
 
11
  # Load the Whisper model once during startup
12
  # device = 0 if torch.cuda.is_available() else -1 # Use GPU if available, otherwise CPU
13
  device = -1
14
+ asr_pipeline = pipeline(model="openai/whisper-tiny", device=device) # Initialize Whisper model
15
 
16
  # Basic GET endpoint
17
  @app.get("/")