Spaces:
Running
on
Zero
Running
on
Zero
artificialguybr
commited on
Commit
•
aa9bb98
1
Parent(s):
349cabe
Update app.py
Browse files
app.py
CHANGED
@@ -21,6 +21,8 @@ from tqdm import tqdm
|
|
21 |
from numba import jit
|
22 |
|
23 |
os.environ["COQUI_TOS_AGREED"] = "1"
|
|
|
|
|
24 |
|
25 |
ZipFile("ffmpeg.zip").extractall()
|
26 |
st = os.stat('ffmpeg')
|
@@ -82,12 +84,17 @@ def process_video(radio, video, target_language, has_closeup_face):
|
|
82 |
|
83 |
shell_command = f"ffmpeg -y -i {run_uuid}_output_audio.wav -af lowpass=3000,highpass=100 {run_uuid}_output_audio_final.wav".split(" ")
|
84 |
subprocess.run([item for item in shell_command], capture_output=False, text=True, check=True)
|
85 |
-
|
86 |
-
|
87 |
-
|
88 |
-
|
89 |
-
|
90 |
-
|
|
|
|
|
|
|
|
|
|
|
91 |
language_mapping = {'English': 'en', 'Spanish': 'es', 'French': 'fr', 'German': 'de', 'Italian': 'it', 'Portuguese': 'pt', 'Polish': 'pl', 'Turkish': 'tr', 'Russian': 'ru', 'Dutch': 'nl', 'Czech': 'cs', 'Arabic': 'ar', 'Chinese (Simplified)': 'zh-cn'}
|
92 |
target_language_code = language_mapping[target_language]
|
93 |
translator = Translator()
|
@@ -149,6 +156,9 @@ def process_video(radio, video, target_language, has_closeup_face):
|
|
149 |
|
150 |
return output_video_path
|
151 |
|
|
|
|
|
|
|
152 |
|
153 |
def swap(radio):
|
154 |
if(radio == "Upload"):
|
|
|
21 |
from numba import jit
|
22 |
|
23 |
os.environ["COQUI_TOS_AGREED"] = "1"
|
24 |
+
api = HfApi(token=HF_TOKEN)
|
25 |
+
repo_id = "artificialguybr/video-dubbing"
|
26 |
|
27 |
ZipFile("ffmpeg.zip").extractall()
|
28 |
st = os.stat('ffmpeg')
|
|
|
84 |
|
85 |
shell_command = f"ffmpeg -y -i {run_uuid}_output_audio.wav -af lowpass=3000,highpass=100 {run_uuid}_output_audio_final.wav".split(" ")
|
86 |
subprocess.run([item for item in shell_command], capture_output=False, text=True, check=True)
|
87 |
+
try:
|
88 |
+
segments, info = model.transcribe(f"{run_uuid}_output_audio_final.wav", beam_size=5)
|
89 |
+
whisper_text = " ".join(segment.text for segment in segments)
|
90 |
+
whisper_language = info.language
|
91 |
+
print(whisper_text)
|
92 |
+
except RuntimeError as e:
|
93 |
+
if "CUDA failed with error device-side assert triggered" in str(e):
|
94 |
+
gr.Warning("Error. Space need to restart. Please retry in a minute")
|
95 |
+
# Restart the script
|
96 |
+
api.restart_space(repo_id=repo_id)
|
97 |
+
|
98 |
language_mapping = {'English': 'en', 'Spanish': 'es', 'French': 'fr', 'German': 'de', 'Italian': 'it', 'Portuguese': 'pt', 'Polish': 'pl', 'Turkish': 'tr', 'Russian': 'ru', 'Dutch': 'nl', 'Czech': 'cs', 'Arabic': 'ar', 'Chinese (Simplified)': 'zh-cn'}
|
99 |
target_language_code = language_mapping[target_language]
|
100 |
translator = Translator()
|
|
|
156 |
|
157 |
return output_video_path
|
158 |
|
159 |
+
except RuntimeError as e:
|
160 |
+
if "CUDA failed with error device-side assert triggered" in str(e):
|
161 |
+
gr.Warning("Unhandled Exception encountered, please retry in a minute")
|
162 |
|
163 |
def swap(radio):
|
164 |
if(radio == "Upload"):
|