Update app.py

app.py

@@ -26,21 +26,7 @@ headers = {"Authorization": f"Bearer {H_key}"}
 def query_image_generation(payload, max_retries=5):
     for attempt in range(max_retries):
         response = requests.post(API_URL, headers=headers, json=payload)
-
-        if response.status_code == 503:
-            print(f"Model is still loading, retrying... Attempt {attempt + 1}/{max_retries}")
-            estimated_time = min(response.json().get("estimated_time", 60), 60)
-            time.sleep(estimated_time)
-            continue
-
-        if response.status_code != 200:
-            print(f"Error: Received status code {response.status_code}")
-            print(f"Response: {response.text}")
-            return None
-
         return response.content
-
-    print(f"Failed to generate image after {max_retries} attempts.")
     return None
 
 # Function for generating an image from text