NeuralFalcon commited on
Commit
d40ba37
·
verified ·
1 Parent(s): edee58e

Update subtitle.py

Browse files
Files changed (1) hide show
  1. subtitle.py +7 -7
subtitle.py CHANGED
@@ -208,13 +208,13 @@ def whisper_subtitle(uploaded_file, source_language):
208
  # 1. Configure device and model
209
  device = "cuda" if torch.cuda.is_available() else "cpu"
210
  compute_type = "float16" if torch.cuda.is_available() else "int8"
211
- model_dir = download_model(
212
- "deepdml/faster-whisper-large-v3-turbo-ct2",
213
- download_folder="./",
214
- redownload=False
215
- )
216
- model = WhisperModel(model_dir, device=device, compute_type=compute_type)
217
- # model = WhisperModel("deepdml/faster-whisper-large-v3-turbo-ct2",device=device, compute_type=compute_type)
218
 
219
 
220
  # 2. Process audio file
 
208
  # 1. Configure device and model
209
  device = "cuda" if torch.cuda.is_available() else "cpu"
210
  compute_type = "float16" if torch.cuda.is_available() else "int8"
211
+ # model_dir = download_model(
212
+ # "deepdml/faster-whisper-large-v3-turbo-ct2",
213
+ # download_folder="./",
214
+ # redownload=False
215
+ # )
216
+ # model = WhisperModel(model_dir, device=device, compute_type=compute_type)
217
+ model = WhisperModel("deepdml/faster-whisper-large-v3-turbo-ct2",device=device, compute_type=compute_type)
218
 
219
 
220
  # 2. Process audio file