Update main.py

dalle3xl update

main.py

@@ -112,6 +112,7 @@ app = FastAPI(
     docs_url="/"
 )
 
+timeout = 100
 contact_support = """
 We are aware that this service is currently offline. This seems to be caused by the API
 We are investigating and doing our best to get things back online as soon as possible. Thank you for your patience
@@ -268,7 +269,7 @@ def dalle_3xl(
             "strength": item.strength
             }
         headers = {"Authorization": f"Bearer {HUGGING_TOKEN}"}
-        response = requests.post(API_URL, headers=headers, json=payload)
+        response = requests.post(API_URL, headers=headers, json=payload, timeout=timeout)
         response.raise_for_status()
     except requests.exceptions.RequestException:
         raise HTTPException(