Spaces:
Runtime error
Runtime error
Update app/services/image_service.py
Browse files
app/services/image_service.py
CHANGED
|
@@ -219,7 +219,7 @@ def virtual_try_on(
|
|
| 219 |
|
| 220 |
|
| 221 |
|
| 222 |
-
def generate_shoe_images(
|
| 223 |
"""Generate photorealistic shoe images from a raw user prompt."""
|
| 224 |
logger.info("Generating shoe images from prompt...")
|
| 225 |
|
|
@@ -229,7 +229,7 @@ def generate_shoe_images(final_prompt: str) -> Tuple[Optional[str], List[BytesIO
|
|
| 229 |
# Try Gemini First
|
| 230 |
response = genai_client.models.generate_content(
|
| 231 |
model="gemini-2.5-flash-image",
|
| 232 |
-
contents=
|
| 233 |
config=config
|
| 234 |
)
|
| 235 |
|
|
|
|
| 219 |
|
| 220 |
|
| 221 |
|
| 222 |
+
def generate_shoe_images(prompt: str) -> Tuple[Optional[str], List[BytesIO]]:
|
| 223 |
"""Generate photorealistic shoe images from a raw user prompt."""
|
| 224 |
logger.info("Generating shoe images from prompt...")
|
| 225 |
|
|
|
|
| 229 |
# Try Gemini First
|
| 230 |
response = genai_client.models.generate_content(
|
| 231 |
model="gemini-2.5-flash-image",
|
| 232 |
+
contents=prompt,
|
| 233 |
config=config
|
| 234 |
)
|
| 235 |
|