Update app.py

app.py

@@ -28,7 +28,7 @@ def enhance_prompt(prompt, style):
     return response.text
 
 def generate_image(enhanced_prompt, style, negative_prompt):
-    model = genai.GenerativeModel("imagen-3.0-generate-002")
+    model = genai.GenerativeModel("gemini-2.0-flash")
     full_prompt = f"{enhanced_prompt}\nStyle: {style}\nNegative prompt: {negative_prompt}"
     
     response = model.generate_content(full_prompt)