Update app.py

app.py

@@ -46,8 +46,8 @@ def generate_image(model, enhanced_prompt, style, negative_prompt):
 
 def process_and_generate(api_key, prompt, style, negative_prompt):
     genai.configure(api_key=api_key)
-    text_model = genai.GenerativeModel('gemini-pro')
-    image_model = genai.GenerativeModel('gemini-pro-vision')
+    text_model = genai.GenerativeModel('gemini-2.0-flash-lite')
+    image_model = genai.GenerativeModel('imagen-3.0-generate-002')
     enhanced_prompt = enhance_prompt(text_model, prompt, style)
     image_url = generate_image(image_model, enhanced_prompt, style, negative_prompt)
     return image_url, enhanced_prompt