Update app.py

app.py

@@ -66,7 +66,7 @@ login(HF_TOKEN)
 
 SD = InferenceClient(model="stabilityai/stable-diffusion-2-1")
 LLM = InferenceClient(model="openchat/openchat-3.5-0106")
-RF = InferenceClient(model="stabilityai/stable-diffusion-xl-refiner-1.0")
+RF = InferenceClient(model="stabilityai/stable-diffusion-xl-refiner-1.0", wait_)
 
 
 def ec(x, fd="<|image|>", sd="<|image|>"):
@@ -203,6 +203,9 @@ async def on_message(message):
                 image = SD.text_to_image(imgp)
                 image.save("image.png")
                 image = RF.image_to_image("image.png")
+                if image["warnings"]:
+                    print(image)
+                    return 0
                 image.save("image.png")
                 file = discord.File("image.png", filename="image.png")
                 embed.set_image(url="attachment://image.png")