Ankush Rana commited on
Commit
23dbd9b
·
1 Parent(s): d82f8bf

add medium model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ import torch
7
  device = "cuda" if torch.cuda.is_available() else "cpu"
8
  torch_dtype = "float32"
9
 
10
- MODEL_NAME = "Systran/faster-whisper-large-v3"
11
  print("Loading model ...")
12
  model = WhisperModel(MODEL_NAME, compute_type=torch_dtype)
13
  print("Loading model done.")
 
7
  device = "cuda" if torch.cuda.is_available() else "cpu"
8
  torch_dtype = "float32"
9
 
10
+ MODEL_NAME = "Systran/faster-whisper-medium"
11
  print("Loading model ...")
12
  model = WhisperModel(MODEL_NAME, compute_type=torch_dtype)
13
  print("Loading model done.")