Update app.py

app.py

@@ -86,21 +86,9 @@ 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)
     
-    print("FFmpeg command completed.")
-
-    # Execute Whisper transcription
-    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)
+    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
             
     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]