Update app.py

app.py

@@ -25,7 +25,9 @@ def query(prompt, is_negative=False, steps=1, cfg_scale=6, seed=None):
         "seed": seed if seed is not None else random.randint(-1, 2147483647)
     }
 
-    image_bytes = requests.post(API_URL, headers=headers, json=payload).content
+    response = requests.post(API_URL, headers=headers, json=payload)
+    response.raise_for_status()
+    image_bytes = response.content
     image = Image.open(io.BytesIO(image_bytes))
     return image