Update app.py

app.py

@@ -15,16 +15,6 @@ import streamlit as st
 from huggingface_hub import AsyncInferenceClient, login
 import time
 import os
-from gradio_client import Client
-
-async def C(x):
-    c = Client(src="ybelkada/llava-1-5-dlai")
-    print(c.view_api(return_format="dict"))
-    result = c.predict(
-        "Make a detailed description of the image. This would be sufficient for an LLM to answer questions about it with just your answer.",
-        "ip.png",
-        api_name="/predict"
-    )
 
 launch_time = datetime.datetime.utcnow()
 
@@ -94,6 +84,7 @@ PT = AsyncInferenceClient(model=proteus)
 LLM = AsyncInferenceClient(model="openchat/openchat-3.5-0106")
 RF = AsyncInferenceClient(model="stabilityai/stable-diffusion-xl-refiner-1.0")
 UP = AsyncInferenceClient(model="radames/stable-diffusion-x4-upscaler-img2img")
+IC = AsyncInferenceClient(model="Salesforce/blip-image-captioning-large")
 
 
 def ec(x, fd="<|image|>", sd="<|image|>"):
@@ -392,7 +383,8 @@ async def on_message(message):
                     Use relatively short prompts for images(20 words max), but still put details.
                     Do not generate images unless the user speciifes that they want an image.
                     If a user has [bot] next to their username, they are a bot.
-                    If there is ImageParsed stuff at the end of the message, that is the text relative to the image."""
+                    If there is ImageParsed stuff at the end of the message, that is the text relative to the image.
+                    If there is (Replied:[]) stuff at the start of the message, that is the message the user replied to."""
         try:
             os.mkdir("data/" + guild_name)
         except:
@@ -402,7 +394,7 @@ async def on_message(message):
             message.content = f"[Replied to: ({str(message.reference.cached_message.author)}: {message.reference.cached_message.content})]; {message.content}"
         if len(message.attachments) > 0:
             await message.attachments[0].save('ip.png')
-            caption = f"(ImageParsed:[{await C('ip.png')}])"
+            caption = f"(ImageParsed:[{await IC.image_to_text('ip.png')}])"
         if os.path.exists(f"data/{guild_name}/{msgchannel_name}"):
             with open(f"data/{guild_name}/{msgchannel_name}", "a") as f:
                 n = "\n"