ankush13r commited on
Commit
cd6ce77
·
verified ·
1 Parent(s): 9269a20

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,9 +4,9 @@ from AinaTheme import theme
4
  from faster_whisper import WhisperModel
5
  import torch
6
 
7
- device, torch_dtype = ("cuda", "float32") if torch.cuda.is_available() else ("cpu", "int16")
8
 
9
- MODEL_NAME = "Systran/faster-whisper-medium"
10
  print("Loading model ...")
11
  model = WhisperModel(MODEL_NAME, compute_type=torch_dtype)
12
  print("Loading model done.")
 
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 ...")
11
  model = WhisperModel(MODEL_NAME, compute_type=torch_dtype)
12
  print("Loading model done.")