Update app.py

app.py

@@ -14,16 +14,15 @@ from PIL import Image
 
 session = requests.Session()
 
-def _fetch_url(*args, **kwargs):
-    for i in range(5):
+def _fetch_url(*args, retries=5, **kwargs):
+    for i in range(retries + 1):
         try:
             response = session.get(*args, **kwargs)
             response.raise_for_status()
             return response.content
-        except requests.RequestException as e:
+        except requests.RequestException:
             if response.status_code in {429, 500, 502, 503, 504}:
                 delay = min(2 ** i, 60)
-                gr.Warning(f"Retrying in {delay:.2f}s..")
                 time.sleep(delay)
             else:
                 return None