Carlexx commited on
Commit
9dd5e68
Β·
verified Β·
1 Parent(s): f7c5a08

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -155,7 +155,7 @@ def get_static_scenes_storyboard(num_fragments: int, prompt: str, initial_image_
155
  prompt_file = "prompts/photographer_prompt.txt"
156
  with open(os.path.join(os.path.dirname(__file__), prompt_file), "r", encoding="utf-8") as f: template = f.read()
157
  director_prompt = template.format(user_prompt=prompt, num_fragments=int(num_fragments))
158
- model = genai.GenerativeModel('gemini-1.5-flash'); img = Image.open(initial_image_path)
159
  response = model.generate_content([director_prompt, img])
160
  try:
161
  cleaned_response = response.text.strip().replace("```json", "").replace("```", "")
@@ -203,7 +203,7 @@ def get_single_motion_prompt(user_prompt: str, story_history: str, start_image_p
203
  if not GEMINI_API_KEY: raise gr.Error("Chave da API Gemini nΓ£o configurada!")
204
  try:
205
  genai.configure(api_key=GEMINI_API_KEY)
206
- model = genai.GenerativeModel('gemini-1.5-flash')
207
  start_img, middle_img, end_img = Image.open(start_image_path), Image.open(middle_image_path), Image.open(end_image_path)
208
  prompt_file_path = os.path.join(os.path.dirname(__file__), "prompts", "director_motion_prompt_three_act.txt")
209
  with open(prompt_file_path, "r", encoding="utf-8") as f:
 
155
  prompt_file = "prompts/photographer_prompt.txt"
156
  with open(os.path.join(os.path.dirname(__file__), prompt_file), "r", encoding="utf-8") as f: template = f.read()
157
  director_prompt = template.format(user_prompt=prompt, num_fragments=int(num_fragments))
158
+ model = genai.GenerativeModel('gemini-2.0-flash'); img = Image.open(initial_image_path)
159
  response = model.generate_content([director_prompt, img])
160
  try:
161
  cleaned_response = response.text.strip().replace("```json", "").replace("```", "")
 
203
  if not GEMINI_API_KEY: raise gr.Error("Chave da API Gemini nΓ£o configurada!")
204
  try:
205
  genai.configure(api_key=GEMINI_API_KEY)
206
+ model = genai.GenerativeModel('gemini-2.p0-flash')
207
  start_img, middle_img, end_img = Image.open(start_image_path), Image.open(middle_image_path), Image.open(end_image_path)
208
  prompt_file_path = os.path.join(os.path.dirname(__file__), "prompts", "director_motion_prompt_three_act.txt")
209
  with open(prompt_file_path, "r", encoding="utf-8") as f: