sushilideaclan01 commited on
Commit
b35d339
·
1 Parent(s): 4ed4ca0

change the correction model

Browse files
frontend/components/generation/CorrectionModal.tsx CHANGED
@@ -117,7 +117,7 @@ export const CorrectionModal: React.FC<CorrectionModalProps> = ({
117
  case "correcting":
118
  return "Generating corrections...";
119
  case "regenerating":
120
- return "Regenerating with nano-banana-pro...";
121
  case "complete":
122
  return "Correction complete!";
123
  case "error":
 
117
  case "correcting":
118
  return "Generating corrections...";
119
  case "regenerating":
120
+ return "Regenerating with image...";
121
  case "complete":
122
  return "Correction complete!";
123
  case "error":
services/correction.py CHANGED
@@ -313,7 +313,7 @@ Respond with valid JSON only, no markdown formatting."""
313
 
314
  image_bytes, model_used, image_url = await image_service.generate(
315
  prompt=focused_prompt,
316
- model_key="nano-banana-pro", # Always use nano-banana-pro for corrections
317
  width=width,
318
  height=height,
319
  image_url=original_image_url, # Pass original image URL for image-to-image
 
313
 
314
  image_bytes, model_used, image_url = await image_service.generate(
315
  prompt=focused_prompt,
316
+ model_key="nano-banana", # Always use nano-banana for corrections
317
  width=width,
318
  height=height,
319
  image_url=original_image_url, # Pass original image URL for image-to-image