Update app.py

app.py

@@ -103,7 +103,7 @@ class NotebookLMTool:
                 """
 
                 resp_img = self.client.models.generate_content(
-                    model="gemini-2.0-flash-exp", 
+                    model="gemini-2.5-flash-image", 
                     contents=[clean_prompt, img],
                     config=types.GenerateContentConfig(response_modalities=["IMAGE"])
                 )
@@ -152,7 +152,7 @@ class NotebookLMTool:
                 """
                 
                 resp_layout = self.client.models.generate_content(
-                    model="gemini-2.0-flash", 
+                    model="gemini-2.5-flash", 
                     contents=[layout_prompt, img],
                     config=types.GenerateContentConfig(response_mime_type="application/json")
                 )