ankush13r commited on
Commit
4751cf9
·
verified ·
1 Parent(s): 8300d85

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -4,8 +4,7 @@ from AinaTheme import theme
4
  from faster_whisper import WhisperModel
5
  import torch
6
 
7
- device = "cuda" if torch.cuda.is_available() else "cpu"
8
- torch_dtype = "int8"
9
 
10
  MODEL_NAME = "Systran/faster-whisper-large-v3"
11
  print("Loading model ...")
 
4
  from faster_whisper import WhisperModel
5
  import torch
6
 
7
+ device, torch_dtype = "cuda", "float32" if torch.cuda.is_available() else "cpu", "int8"
 
8
 
9
  MODEL_NAME = "Systran/faster-whisper-large-v3"
10
  print("Loading model ...")