Update main.py

main.py

@@ -8,6 +8,7 @@ import base64
 import requests
 import os
 import socket
+import time
 
 #--------------------------------------------------- Definizione Server FAST API ------------------------------------------------------
 app = FastAPI()
@@ -98,6 +99,7 @@ def generate_image(request: Request, input_data: InputImage):
             return {"response": img_base64}
         except requests.exceptions.HTTPError as e:
             if e.response.status_code == 500:
+                 time.sleep(1)
                 attempt += 1
                 if attempt < max_attempts:
                     continue