Update app.py

app.py

@@ -21,6 +21,8 @@ from tqdm import tqdm
 from numba import jit
 
 os.environ["COQUI_TOS_AGREED"] = "1"
+api = HfApi(token=HF_TOKEN)
+repo_id = "artificialguybr/video-dubbing"
 
 ZipFile("ffmpeg.zip").extractall()
 st = os.stat('ffmpeg')
@@ -82,12 +84,17 @@ def process_video(radio, video, target_language, has_closeup_face):
 
     shell_command = f"ffmpeg -y -i {run_uuid}_output_audio.wav -af lowpass=3000,highpass=100 {run_uuid}_output_audio_final.wav".split(" ")
     subprocess.run([item for item in shell_command], capture_output=False, text=True, check=True)
-
-    segments, info = model.transcribe(f"{run_uuid}_output_audio_final.wav", beam_size=5)
-    whisper_text = " ".join(segment.text for segment in segments)
-    whisper_language = info.language
-    print(whisper_text)
-    
+    try:
+        segments, info = model.transcribe(f"{run_uuid}_output_audio_final.wav", beam_size=5)
+        whisper_text = " ".join(segment.text for segment in segments)
+        whisper_language = info.language
+        print(whisper_text)
+    except RuntimeError as e:
+        if "CUDA failed with error device-side assert triggered" in str(e):
+            gr.Warning("Error. Space need to restart. Please retry in a minute")
+            # Restart the script
+            api.restart_space(repo_id=repo_id)
+            
     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'}
     target_language_code = language_mapping[target_language]
     translator = Translator()
@@ -149,6 +156,9 @@ def process_video(radio, video, target_language, has_closeup_face):
 
     return output_video_path
     
+    except RuntimeError as e:
+        if "CUDA failed with error device-side assert triggered" in str(e):
+            gr.Warning("Unhandled Exception encountered, please retry in a minute")
     
 def swap(radio):
     if(radio == "Upload"):