Update app.py

app.py

@@ -8,13 +8,23 @@ from PIL import Image
 MAX_SEED = 2 ** 31 - 1
 session = requests.Session()
 
+import requests
+from PIL import Image
+
+session = requests.Session()
+
 def _fetch_url(*args, **kwargs):
-    try:
-        response = session.get(*args, **kwargs)
-        response.raise_for_status()
-        return response.content
-    except requests.RequestException as e:
-        raise gr.Error(f"Request error. Status code: {response.status_code}") from None
+    for i in range(5):
+        try:
+            response = session.get(*args, **kwargs)
+            response.raise_for_status()
+            return response.content
+        except requests.RequestException as e:
+            if response.status_code in {429, 500, 502, 503, 504}:
+                delay = min(2 ** i, 60)
+                time.sleep(delay)
+            else:
+                return None
     return None
 
 def add_to_gallery(image, gallery=None):