Update app.py

app.py

@@ -78,6 +78,7 @@ SDT = AsyncInferenceClient(model=sdxl_turbo)
 PT = AsyncInferenceClient(model=proteus)
 LLM = AsyncInferenceClient(model="openchat/openchat-3.5-0106")
 RF = AsyncInferenceClient(model="stabilityai/stable-diffusion-xl-refiner-1.0")
+C = AsyncInferenceClient(model="llava-hf/llava-1.5-7b-hf")
 
 
 def ec(x, fd="<|image|>", sd="<|image|>"):
@@ -146,7 +147,6 @@ async def on_message(message):
             return
         sysp = f"""GPT4 Correct System:
             Instructions for Assistant:
-                You are not the Decider. Do not use the <|res|> or <|nrs|> tags.
                 You are lyre, a discord bot. You were made by Araeyn.
                 Your username is lyre#9828.
                 You must use the markdown format for your responses.
@@ -158,12 +158,7 @@ async def on_message(message):
                 For example, if the user asks you to generate an image of a cat, you could say <|image|>A cute cat with long fur that is looking out a window with curious eyes, hd, realistic<|image|>
                 Always include a title.
                 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.
-            Instructions for Decider:
-                You are Decider, a classifier, not the Assistant.
-                You will see the conversation, and decide whether or not the Assistant should say something in the conversation.
-                If the user mentions lyre#9828 or is trying to speak to you or continue a conversation with you, say <|res|>.
-                If you do, say <|res|>, and if you do not, say <|nrs|>. Do not say anything after."""
+                If a user has [bot] next to their username, they are a bot."""
         try:
             os.mkdir("data/" + guild_name)
         except:
@@ -191,11 +186,7 @@ async def on_message(message):
                     )
         with open(f"data/{guild_name}/{msgchannel_name}", "r") as f:
             context = f.read()
-        res = await LLM.text_generation(context + "<|end_of_turn|>GPT4 Correct Decider: ",
-                                             stop_sequences=["<|end_of_turn|>"],
-                                             max_new_tokens=48)
-        print(res)
-        if ("<|res|>" in res and "<|nrs|>" not in res) or guild_name == "Direct":
+        if ("Feeway" in guild_name) or guild_name == "Direct":
             async with msgchannel.typing():
                 context += f"GPT4 Correct Assistant:"
                 load = random.choice(
@@ -235,7 +226,7 @@ async def on_message(message):
                     text=
                     """Refining..."""
                 )
-                file = discord.File("image.png", filename="image.png")
+                file = discord.File("image.png", filename="image.png", description=imgp)
                 embed.set_image(url="attachment://image.png")
                 await e.edit(embed=embed, attachments=[file])
                 image = await RF.image_to_image("image.png")