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