Update app.py

app.py

@@ -1,14 +1,20 @@
 from langchain_google_genai import ChatGoogleGenerativeAI
 from google import genai
+from google.api_core import retry
 from io import BytesIO
 import gradio as gr
 import base64
 import time
 import uuid
 import os
-"""
-For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
-"""
+
+# Define a retry policy
+is_retriable = lambda e: (isinstance(e, genai.errors.APIError) and e.code in {429, 503})
+
+if not hasattr(genai.models.Models.generate_content, '__wrapped__'):
+  genai.models.Models.generate_content = retry.Retry(
+      predicate=is_retriable)(genai.models.Models.generate_content)
+
 os.environ["GOOGLE_API_KEY"] = os.getenv("GOOGLE_API_KEY")
 
 vis_llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash-preview-image-generation")
@@ -107,4 +113,4 @@ with gr.Blocks() as app:
 
 if __name__ == "__main__":
     app.queue() 
-    app.launch()
+    app.launch(share=True)