Athspi commited on
Commit
43ee34f
·
verified ·
1 Parent(s): 99753e8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -10
app.py CHANGED
@@ -5,7 +5,6 @@ from faster_whisper import WhisperModel
5
  from moviepy.video.io.VideoFileClip import VideoFileClip
6
  import logging
7
  import google.generativeai as genai
8
- from google.api_core import exceptions
9
 
10
  # Suppress moviepy logs
11
  logging.getLogger("moviepy").setLevel(logging.ERROR)
@@ -23,7 +22,7 @@ generation_config = {
23
  }
24
 
25
  model = genai.GenerativeModel(
26
- model_name="gemini-2.0-flash-exp",
27
  generation_config=generation_config,
28
  )
29
 
@@ -98,14 +97,9 @@ def format_timestamp(seconds):
98
 
99
  def translate_text(text, target_language):
100
  """Translate text using Gemini."""
101
- try:
102
- prompt = f"Translate the following text into {target_language}:\n\n{text}"
103
- response = model.generate_content(prompt)
104
- return response.text
105
- except exceptions.ResourceExhausted:
106
- return f"Translation failed: API quota exceeded. Please try again later."
107
- except Exception as e:
108
- return f"Translation failed: {str(e)}"
109
 
110
  def translate_srt(srt_text, target_language):
111
  """Translate an SRT file while preserving timestamps."""
 
5
  from moviepy.video.io.VideoFileClip import VideoFileClip
6
  import logging
7
  import google.generativeai as genai
 
8
 
9
  # Suppress moviepy logs
10
  logging.getLogger("moviepy").setLevel(logging.ERROR)
 
22
  }
23
 
24
  model = genai.GenerativeModel(
25
+ model_name="gemini-exp-1206", # Updated model name
26
  generation_config=generation_config,
27
  )
28
 
 
97
 
98
  def translate_text(text, target_language):
99
  """Translate text using Gemini."""
100
+ prompt = f"Translate the following text into {target_language}:\n\n{text}"
101
+ response = model.generate_content(prompt)
102
+ return response.text
 
 
 
 
 
103
 
104
  def translate_srt(srt_text, target_language):
105
  """Translate an SRT file while preserving timestamps."""