Update subtitle.py
Browse files- 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 |
-
|
| 213 |
-
|
| 214 |
-
|
| 215 |
-
)
|
| 216 |
-
model = WhisperModel(model_dir, device=device, compute_type=compute_type)
|
| 217 |
-
|
| 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
|