Update app.py

app.py

@@ -75,13 +75,13 @@ model = "openchat/openchat-3.5-0106"
 API_URL = "https://api-inference.huggingface.co/models/" + model
 headers = {"Authorization": f"Bearer {HF_TOKEN}"}
 
-from diffusers import StableDiffusionXLImg2ImgPipeline
-from diffusers.utils import load_image
+# from diffusers import StableDiffusionXLImg2ImgPipeline
+# from diffusers.utils import load_image
 
-pipe = StableDiffusionXLImg2ImgPipeline.from_pretrained(
-    "stabilityai/stable-diffusion-xl-refiner-1.0", torch_dtype=torch.float32, use_safetensors=True
-)
-pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
+# pipe = StableDiffusionXLImg2ImgPipeline.from_pretrained(
+#    "stabilityai/stable-diffusion-xl-refiner-1.0", torch_dtype=torch.float32, use_safetensors=True
+# )
+# pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
 
 def generate_latex_image(latex_code):
   api_url = "https://latex.codecogs.com/svg.latex?"
@@ -103,6 +103,12 @@ def query(x):
   Response = requests.post(API_URL, headers=headers, json=x)
   return Response.json()
 
+def caption(filename):
+    with open(filename, "rb") as f:
+        data = f.read()
+    response = requests.post(API_URL, headers=headers, data=data)
+    return response.json()
+
 
 launch_time = datetime.datetime.utcnow()
 
@@ -152,7 +158,11 @@ async def on_message(message:discord.Message):
     print(s)
     if message.author == client.user:
       return
-    s = f"GPT4 Correct {message.author}: {message.content}<|end_of_turn|>"
+    ats = []
+    if len(message.attachments) > 0:
+        for i in message.attachments:
+            ats.append(f"ImageInput:{i}")
+    s = f"GPT4 Correct {message.author}: {message.content} Images: {" ".join(ats)}<|end_of_turn|>"
     try:
       with open(os.path.join(guild_name, f"{msgchannel_name}.txt"), "a") as f:
         f.write(s)