Update app.py

app.py

@@ -25,9 +25,14 @@ def generate_image(prompt):
   prompt = tuple(prompt)
 
   response = client.models.generate_content(
-    model="gemini-2.5-flash-image-preview",
-    contents=[prompt],
-)
+    model="gemini-2.0-flash-preview-image-generation",
+    contents=prompt,
+    config=types
+    .GenerateContentConfig(
+      response_modalities=['TEXT', 'IMAGE']
+    )
+  )
+    
   for part in response.candidates[0].content.parts:
     if part.text is not None:
       pass