Update main.py

main.py

@@ -104,7 +104,7 @@ def generate_and_upload_image(prompt: str) -> str:
     """
     try:
         response = genai_client.models.generate_content(
-            model="gemini-2.0-flash-exp",
+            model="gemini-2.0-flash-preview-image-generation",
             contents=prompt,
             config=types.GenerateContentConfig(response_modalities=['Text', 'Image'])
         )
@@ -147,7 +147,7 @@ def analyze_image_with_text(image_path: str, user_text: str) -> str:
         img_user = PIL.Image.open(image_path)
         print(user_text)
         response = genai_client.models.generate_content(
-            model="gemini-2.0-flash-exp",
+            model="gemini-2.0-flash-preview-image-generation",
             contents=[img_user, user_text]
         )