Spaces:
Sleeping
Sleeping
Rename ai_studio_code (13).py to app.py
Browse files
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-
|
| 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-
|
| 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)
|