VeuReu commited on
Commit
f195d1e
·
verified ·
1 Parent(s): 1a6850f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -56,8 +56,7 @@ def _lazy_load_whisper():
56
  use_safetensors=True
57
  )
58
 
59
- device = "cuda" if torch.cuda.is_available() else "cpu"
60
- m = m.to(device)
61
 
62
  _model_whis = m
63
 
 
56
  use_safetensors=True
57
  )
58
 
59
+ m = m.to(DEVICE)
 
60
 
61
  _model_whis = m
62