Update config.py

config.py

@@ -10,8 +10,8 @@ load_dotenv()
 GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
 
 # Model Configuration
-VLM_MODEL = "gemini-3-pro-preview"  # For Retriever, Planner, Stylist, Critic
-IMAGE_MODEL = "gemini-3-pro-image-preview"  # For Visualizer (referred to as Nano-Banana-Pro in paper)
+VLM_MODEL = "gemini-2.0-flash"  # For Retriever, Planner, Stylist, Critic
+IMAGE_MODEL = "gemini-2.0-flash"  # For Visualizer (referred to as Nano-Banana-Pro in paper)
 
 # Generation Configuration
 MAX_REFINEMENT_ITERATIONS = 3  # As per ablation study