Update app.py

app.py

@@ -8,7 +8,6 @@ from threading import Thread
 import discord
 from discord.ext import tasks
 from PIL import Image
-from colorama import Fore, Style
 import streamlit as st
 from huggingface_hub import InferenceClient, login
 import time
@@ -66,6 +65,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")
 
 
 def ec(x, fd="<image>", sd="</image>"):
@@ -195,14 +195,15 @@ async def on_message(message):
                 embed.set_image(url=load)
                 image = SD.text_to_image(imgp)
                 image.save("image.png")
-                file = discord.File("image.png")
+                image = RF.image_to_image("image.png")
+                image.save("image.png")
+                file = discord.File("image.png", filename="image.png")
                 embed.set_image(url="attachment://image.png")
                 e.edit(embed=embed, file=file)
+                
             
     except Exception as e:
-        print(Fore.RED)
         print(e)
-        print(Style.RESET_ALL)
 
 token = os.environ["TOKEN"]
 client.run(token)