DeeeeeeM
commited on
Commit
·
77b6231
1
Parent(s):
0f42c39
removed batched inference feature temporarily due to unknown issues
Browse files- app.py +4 -6
- requirements-gpu.txt +8 -0
- requirements.txt +2 -1
app.py
CHANGED
|
@@ -25,16 +25,14 @@ def process_media(
|
|
| 25 |
if model_type == "faster whisper":
|
| 26 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 27 |
model = stable_whisper.load_faster_whisper(model_size, device=device)
|
|
|
|
| 28 |
else:
|
| 29 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 30 |
model = stable_whisper.load_model(model_size, device=device)
|
|
|
|
|
|
|
|
|
|
| 31 |
|
| 32 |
-
try:
|
| 33 |
-
result = model.transcribe(temp_path, language=source_lang, vad=True, regroup=False, no_speech_threshold=0.9, denoiser="demucs", batch_size=16)
|
| 34 |
-
#result.save_as_json(word_transcription_path)
|
| 35 |
-
except Exception as e:
|
| 36 |
-
return None, None, None, None
|
| 37 |
-
|
| 38 |
# ADVANCED SETTINGS #
|
| 39 |
if max_chars or max_words:
|
| 40 |
result.split_by_length(
|
|
|
|
| 25 |
if model_type == "faster whisper":
|
| 26 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 27 |
model = stable_whisper.load_faster_whisper(model_size, device=device)
|
| 28 |
+
result = model.transcribe(temp_path, language=source_lang, vad=True, regroup=False, no_speech_threshold=0.9, denoiser="demucs")
|
| 29 |
else:
|
| 30 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 31 |
model = stable_whisper.load_model(model_size, device=device)
|
| 32 |
+
result = model.transcribe(temp_path, language=source_lang, vad=True, regroup=False, no_speech_threshold=0.9, denoiser="demucs")
|
| 33 |
+
#, batch_size=16
|
| 34 |
+
#result.save_as_json(word_transcription_path)
|
| 35 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 36 |
# ADVANCED SETTINGS #
|
| 37 |
if max_chars or max_words:
|
| 38 |
result.split_by_length(
|
requirements-gpu.txt
ADDED
|
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
gradio
|
| 2 |
+
stable-ts
|
| 3 |
+
stable-ts[fw]
|
| 4 |
+
torch==2.6.0+cu124
|
| 5 |
+
numpy<2.3,>=1.24
|
| 6 |
+
sympy==1.13.1
|
| 7 |
+
demucs
|
| 8 |
+
chardet
|
requirements.txt
CHANGED
|
@@ -3,5 +3,6 @@ stable-ts
|
|
| 3 |
stable-ts[fw]
|
| 4 |
demucs
|
| 5 |
torch==2.6.0
|
| 6 |
-
numpy
|
|
|
|
| 7 |
chardet
|
|
|
|
| 3 |
stable-ts[fw]
|
| 4 |
demucs
|
| 5 |
torch==2.6.0
|
| 6 |
+
numpy>=1.24,<2.3
|
| 7 |
+
sympy==1.13.1
|
| 8 |
chardet
|