unknown user commited on
Commit
56a920f
·
1 Parent(s): 910f5b3
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,9 +3,9 @@ from AinaTheme import theme
3
  from faster_whisper import WhisperModel
4
  import torch
5
 
6
- device, torch_dtype = ("cuda", "float32") if torch.cuda.is_available() else ("cpu", "int8")
7
 
8
- MODEL_NAME = "Systran/faster-whisper-large-v3"
9
  print("Loading model ...")
10
  model = WhisperModel(MODEL_NAME, compute_type=torch_dtype)
11
  print("Loading model done.")
 
3
  from faster_whisper import WhisperModel
4
  import torch
5
 
6
+ device, torch_dtype = ("cuda", "float32") if torch.cuda.is_available() else ("cpu", "float32")
7
 
8
+ MODEL_NAME = "suzii/vi-whisper-large-v3-turbo-v1-ct2"
9
  print("Loading model ...")
10
  model = WhisperModel(MODEL_NAME, compute_type=torch_dtype)
11
  print("Loading model done.")