Update app.py

app.py

@@ -20,18 +20,21 @@ def query(payload, model):
     return response.content
 
 def hf_inference(prompt, negative, model, steps, sampler, guidance, width, height, seed):
-    image_bytes = query(payload={
-        "inputs": f"{prompt}",
-        "parameters": {
-            "negative_prompt": f"{negative}",
-            "num_inference_steps": steps,
-            "guidance_scale": guidance,
-            "width": width, "height": height,
-            "seed": seed,
-        },
-    }, model=model)
-    image = Image.open(io.BytesIO(image_bytes))
-    return image
+    try:
+        image_bytes = query(payload={
+            "inputs": f"{prompt}",
+            "parameters": {
+                "negative_prompt": f"{negative}",
+                "num_inference_steps": steps,
+                "guidance_scale": guidance,
+                "width": width, "height": height,
+                "seed": seed,
+            },
+        }, model=model)
+        image = Image.open(io.BytesIO(image_bytes))
+        return image
+    expect PIL.UnidentifiedImageError:
+        gr.Warning("This model is not loaded now. Try others models.")