wait is this gonna fix it?

elouangrimm · Oct 29, 2024 · 0563cc2 · 0563cc2
1 parent b8df05f
commit 0563cc2
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/bot.py b/bot.py
@@ -95,9 +95,7 @@ async def on_message(message):
         api_url = "https://api-inference.huggingface.co/models/microsoft/DialoGPT-small"
         headers = {"Authorization": f"Bearer {token}"}
 
-        custom_instructions = ai_instructions
-        input_text = custom_instructions + message.content
-
+        input_text = f"{ai_instructions}\nUser: {message.content}\nAI:"
         data = {"inputs": input_text}
         response = requests.post(api_url, headers=headers, json=data)