Update app.py

app.py

@@ -75,6 +75,13 @@ model = "openchat/openchat-3.5-1210"
 API_URL = "https://api-inference.huggingface.co/models/" + model
 headers = {"Authorization": f"Bearer {HF_TOKEN}"}
 
+import torch
+from diffusers import StableDiffusionXLImg2ImgPipeline
+from diffusers.utils import load_image
+
+pipe = StableDiffusionXLImg2ImgPipeline.from_pretrained(
+    "stabilityai/stable-diffusion-xl-refiner-1.0", torch_dtype=torch.float16, variant="fp16", use_safetensors=True
+)
 
 def generate_latex_image(latex_code):
   api_url = "https://latex.codecogs.com/svg.latex?"
@@ -222,6 +229,11 @@ async def on_message(message:discord.Message):
         try: image = Image.open(io.BytesIO(image_bytes))
         except: print(image_bytes)
         image.save(f"latest.png")
+        prompt = ""
+        init_image = load_image(url).convert("RGB")
+        image = pipe(prompt, image=init_image).images[0]
+        pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
+        image.save("latest.png")
         embed.set_image(url="attachment://latest.png")
         await e.edit(embed=embed, attachments=[discord.File(fp=f"latest.png")])
         os.system("rm " + f"latest.png")