s

app.py

@@ -34,7 +34,7 @@ async def on_message(message):
             messages.append(message.content)
 
         previous_messages = ("\n".join(messages))
-        response = generate(previous_messages, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0)
+        response = ai.text_generation(previous_messages, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0)
 
         splited_response = split_string(response)
         for part in splited_response: