s

app.py

@@ -42,8 +42,8 @@ async def on_message(message):
         message_counts[message.channel.id] = 0 # Reset the counter
 
 
-    response = ai.text_generation(str(message), temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0)
-    splited_response = split_string(response)
+    response = generate(str(message), temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0)
+    splited_response = split_string(str(response))
     for part in splited_response:
         await channel.send(part)
 
@@ -51,7 +51,6 @@ async def on_message(message):
 
 def split_string(text: str) -> list[str]:
     """Helper function to split text into chunks"""
-
     return [text[i:i+3900] for i in range(0, len(text), 3900)]
 
 def generate(