GoodML commited on
Commit
ff0d4f5
·
verified ·
1 Parent(s): b6e7946

Added CPU for whisper specifically

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -180,7 +180,7 @@ GEMINI_API_KEY = API_KEY
180
 
181
  # Load Whisper AI model at startup
182
  print("Loading Whisper AI model..., ANIKET")
183
- whisper_model = whisper.load_model("base") # Choose model size: tiny, base, small, medium, large
184
  print("Whisper AI model loaded successfully, ANIKET")
185
 
186
 
 
180
 
181
  # Load Whisper AI model at startup
182
  print("Loading Whisper AI model..., ANIKET")
183
+ whisper_model = whisper.load_model("base", device="CPU") # Choose model size: tiny, base, small, medium, large
184
  print("Whisper AI model loaded successfully, ANIKET")
185
 
186