Carlexxx commited on
Commit
0476445
·
verified ·
1 Parent(s): 1f7ad84

Rename ai_studio_code (13).py to app.py

Browse files
Files changed (1) hide show
  1. ai_studio_code (13).py → app.py +2 -2
ai_studio_code (13).py → app.py RENAMED
@@ -110,7 +110,7 @@ def get_static_scenes_storyboard(num_fragments: int, prompt: str, initial_image_
110
  prompt_file = "prompts/photographer_prompt.txt"
111
  with open(os.path.join(os.path.dirname(__file__), prompt_file), "r", encoding="utf-8") as f: template = f.read()
112
  director_prompt = template.format(user_prompt=prompt, num_fragments=int(num_fragments))
113
- model = genai.GenerativeModel('gemini-1.5-flash'); img = Image.open(initial_image_path)
114
  response = model.generate_content([director_prompt, img])
115
  try:
116
  cleaned_response = response.text.strip().replace("```json", "").replace("```", "")
@@ -160,7 +160,7 @@ def get_motion_storyboard(user_prompt: str, keyframe_image_paths: list):
160
  director_prompt = template.format(user_prompt=user_prompt, num_fragments=len(keyframe_image_paths))
161
  model_contents = [director_prompt]
162
  for img_path in keyframe_image_paths: img = Image.open(img_path); model_contents.append(img)
163
- model = genai.GenerativeModel('gemini-1.5-flash'); response = model.generate_content(model_contents)
164
  try:
165
  cleaned_response = response.text.strip().replace("```json", "").replace("```", "")
166
  storyboard_data = json.loads(cleaned_response)
 
110
  prompt_file = "prompts/photographer_prompt.txt"
111
  with open(os.path.join(os.path.dirname(__file__), prompt_file), "r", encoding="utf-8") as f: template = f.read()
112
  director_prompt = template.format(user_prompt=prompt, num_fragments=int(num_fragments))
113
+ model = genai.GenerativeModel('gemini-2.0-flash'); img = Image.open(initial_image_path)
114
  response = model.generate_content([director_prompt, img])
115
  try:
116
  cleaned_response = response.text.strip().replace("```json", "").replace("```", "")
 
160
  director_prompt = template.format(user_prompt=user_prompt, num_fragments=len(keyframe_image_paths))
161
  model_contents = [director_prompt]
162
  for img_path in keyframe_image_paths: img = Image.open(img_path); model_contents.append(img)
163
+ model = genai.GenerativeModel('gemini-2.0-flash'); response = model.generate_content(model_contents)
164
  try:
165
  cleaned_response = response.text.strip().replace("```json", "").replace("```", "")
166
  storyboard_data = json.loads(cleaned_response)