Hammad712 commited on
Commit
5395a25
·
verified ·
1 Parent(s): 51d915e

Update app/services/image_service.py

Browse files
Files changed (1) hide show
  1. app/services/image_service.py +2 -2
app/services/image_service.py CHANGED
@@ -219,7 +219,7 @@ def virtual_try_on(
219
 
220
 
221
 
222
- def generate_shoe_images(final_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,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=final_prompt,
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