Update app.py

app.py

@@ -131,6 +131,9 @@ def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM+
     if response.status_code != 200:
         print(f"Ошибка: Не удалось получить изображение. Статус ответа: {response.status_code}")
         print(f"Содержимое ответа: {response.text}")
+        if response.status_code == 503:
+            raise gr.Error(f"{response.status_code} : Server overloaded")
+            return None
         raise gr.Error(f"{response.status_code}")
         return None