Update app.py

app.py

@@ -1,11 +1,10 @@
 import gradio as gr
 import whisper
 import moviepy.editor as mp
-from googletrans import Translator
+from deep_translator import GoogleTranslator
 from pydub import AudioSegment
 import os
 import tempfile
-import spaces
 
 def extract_audio(video_path):
     video = mp.VideoFileClip(video_path)
@@ -14,17 +13,16 @@ def extract_audio(video_path):
     audio.write_audiofile(audio_path)
     return audio_path
 
-@spaces.GPU
 def generate_subtitles(audio_path):
     model = whisper.load_model("base")
     result = model.transcribe(audio_path)
     return result["segments"]
 
 def translate_subtitles(subtitles, target_language):
-    translator = Translator()
+    translator = GoogleTranslator(source='auto', target=target_language)
     translated_subtitles = []
     for segment in subtitles:
-        translated_text = translator.translate(segment["text"], dest=target_language).text
+        translated_text = translator.translate(segment["text"])
         translated_subtitles.append({
             "start": segment["start"],
             "end": segment["end"],
@@ -68,11 +66,11 @@ iface = gr.Interface(
     fn=gradio_interface,
     inputs=[
         gr.Video(label="Upload Video"),
-        gr.Dropdown(choices=["es", "fr", "de", "it", "ja", "ko", "zh-cn"], label="Target Language")
+        gr.Dropdown(choices=["es", "fr", "de", "it", "ja", "ko", "zh-CN"], label="Target Language")
     ],
     outputs=gr.Video(label="Processed Video"),
     title="Video Subtitle Translator",
     description="Upload a video, and get it back with translated subtitles!"
 )
 
-iface.launch()
+iface.launch()