fix timeout when waiting for stable diffusion server

fair.py

@@ -149,11 +149,12 @@ def wait_for_server(retries, timeout, delay=1.0):
             r = requests.get(ENDPOINT_ADDRESS, timeout=timeout)
             r.raise_for_status()
             return
-        except (requests.exceptions.ConnectionError, requests.exceptions.HTTPError, requests.exceptions.Timeout):
-            logger.info("Server is not ready yet")
-            time.sleep(delay)
-    else:
-        raise ServerNotReadyException("Failed to start the server")
+        except (requests.exceptions.ConnectionError, requests.exceptions.HTTPError, requests.exceptions.Timeout) as e:
+            if i == retries - 1:
+                raise ServerNotReadyException("Failed to start the server") from e
+            else:
+                logger.info("Server is not ready yet")
+                time.sleep(delay)
 
 
 def start_server():
@@ -186,7 +187,7 @@ def start_server():
 
 def text_to_image(text):
     try:
-        wait_for_server(retries=1, timeout=0.1, delay=0.0)
+        wait_for_server(retries=1, timeout=1.0, delay=0.0)
     except ServerNotReadyException:
         start_server()