IT WILL WORK IM SURE

elouangrimm · Oct 29, 2024 · 7566b37 · 7566b37
1 parent 0563cc2
commit 7566b37
Showing 1 changed file with 15 additions and 5 deletions.
diff --git a/bot.py b/bot.py
@@ -92,11 +92,20 @@ async def on_message(message):
     if bot.user.mentioned_in(message):
         print("Bot Pinged - Attempting Response")
         token = os.getenv("HUGGING_FACE_TOKEN")
-        api_url = "https://api-inference.huggingface.co/models/microsoft/DialoGPT-small"
+        api_url = "https://api-inference.huggingface.co/models/microsoft/GODEL-v1_1-large-seq2seq"
         headers = {"Authorization": f"Bearer {token}"}
 
-        input_text = f"{ai_instructions}\nUser: {message.content}\nAI:"
-        data = {"inputs": input_text}
+        input_text = f"User: {message.content}\nAI:"
+        data = {
+            "inputs": input_text,
+            "parameters": {
+                "max_new_tokens": 50,
+                "do_sample": True,
+                "top_k": 50,
+                "top_p": 0.95,
+                "temperature": 0.9,
+            }
+        }
         response = requests.post(api_url, headers=headers, json=data)
 
         def handle_error(response):
@@ -107,8 +116,8 @@ def handle_error(response):
             json_data = response.json()
             print(json_data)
 
-            if isinstance(json_data, list) and len(json_data) > 0 and "generated_text" in json_data[0]:
-                reply_text = json_data[0]["generated_text"]
+            if isinstance(json_data, dict) and "generated_text" in json_data:
+                reply_text = json_data["generated_text"]
                 print("AI: Message Generated, Code 200")
             else:
                 reply_text = "Hmm... couldn't quite generate a response! 😅"
@@ -127,6 +136,7 @@ def handle_error(response):
 
     await bot.process_commands(message)
 
+
 # Add Ping Command
 @bot.command()
 async def addping(ctx, role: discord.Role, member: discord.Member = None):