Radosław Wolnik commited on
Commit
56ce1e7
·
1 Parent(s): 74b7d49
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -33,10 +33,11 @@ async def on_message(message):
33
  messages.append(message.content)
34
 
35
  previous_messages = ("\n".join(messages))
 
36
 
37
- x = generate(previous_messages, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0)
38
-
39
- await channel.send(x)
40
 
41
  message_counts[message.channel.id] = 0 # Reset the counter
42
 
 
33
  messages.append(message.content)
34
 
35
  previous_messages = ("\n".join(messages))
36
+ raw = generate(previous_messages, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0)
37
 
38
+ splited = split_string(raw)
39
+ for part in splited:
40
+ await channel.send(part)
41
 
42
  message_counts[message.channel.id] = 0 # Reset the counter
43