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", wait_)
+RF = InferenceClient(model="stabilityai/stable-diffusion-xl-refiner-1.0")
 
 
 def ec(x, fd="<|image|>", sd="<|image|>"):
@@ -198,18 +198,17 @@ async def on_message(message):
                 )
                 e = await message.reply(embed=embed)
             print(imgp)
-            if imgp != "":
-                embed.set_image(url=load)
-                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")
-                await e.edit(embed=embed, attachments=[file])
+            embed.set_image(url=load)
+            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")
+            await e.edit(embed=embed, attachments=[file])
                 
             
     except Exception as e: