Update app.py

app.py

@@ -86,9 +86,18 @@ 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=4)
-    whisper_text = " ".join(segment.text for segment in segments)
-    whisper_language = info.language
+    print("Attempting to transcribe with Whisper...")
+    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(f"Transcription successful: {whisper_text}")
+    except RuntimeError as e:
+        print(f"RuntimeError encountered: {str(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]