Update main.py

main.py

@@ -33,8 +33,8 @@ class InputImage(BaseModel):
     input: str
     negativePrompt: str = '' 
     steps: int = 25
-    cfg: int = 5
-    seed: int = 453666937
+    cfg: int = 7
+    seed: int = -1
 
 #--------------------------------------------------- Generazione TESTO ------------------------------------------------------
 @app.post("/Genera")
@@ -78,7 +78,7 @@ def format_prompt(message, history):
 @app.post("/Immagine")
 def generate_image(request: Request, input_data: InputImage):
     client = Client("https://openskyml-fast-sdxl-stable-diffusion-xl.hf.space/--replicas/545b5tw7n/")
-    max_attempts = 10
+    max_attempts = 20
     attempt = 0
     while attempt < max_attempts:
         try:
@@ -98,15 +98,12 @@ def generate_image(request: Request, input_data: InputImage):
                 img_base64 = base64.b64encode(img_binary).decode('utf-8')
             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
-                else:
-                    return {"error": "Errore interno del server persistente"}
+            time.sleep(1)
+            attempt += 1
+            if attempt < max_attempts:
+                continue
             else:
-                return {"error": "Errore diverso da 500"}
+                return {"error": "Errore interno del server persistente"}
     return {"error": "Numero massimo di tentativi raggiunto"}
 
 @app.get("/")