Charlie Amalet commited on
Commit
2c04385
·
verified ·
1 Parent(s): 5ffc16c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ SAVE_DIR = LOCAL_DIR / "transcripts"
15
  SAVE_DIR.mkdir(exist_ok=True)
16
  LANGS = [lang.capitalize() for lang in list(LANGUAGES.values())]
17
  DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
18
- loaded_model = whisper.load_model("medium", DEVICE)
19
  current_size = "None"
20
 
21
  def generate_random_filename():
 
15
  SAVE_DIR.mkdir(exist_ok=True)
16
  LANGS = [lang.capitalize() for lang in list(LANGUAGES.values())]
17
  DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
18
+ loaded_model = whisper.load_model("medium", "cpu")
19
  current_size = "None"
20
 
21
  def generate_random_filename():