Radosław Wolnik commited on
Commit
044febb
·
1 Parent(s): 4ef87fa

Responding

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -48,7 +48,7 @@ def split_string(text: str) -> list[str]:
48
  return [text[i:i+3900] for i in range(0, len(text), 3900)]
49
 
50
  def generate(
51
- prompt, details=False, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0
52
  ):
53
  temperature = float(temperature)
54
  if temperature < 1e-2:
@@ -64,7 +64,7 @@ def generate(
64
  seed=42,
65
  )
66
 
67
- return ai.text_generation(prompt, **generate_kwargs, stream=False, details=details)
68
 
69
  @bot.event
70
  async def on_ready():
 
48
  return [text[i:i+3900] for i in range(0, len(text), 3900)]
49
 
50
  def generate(
51
+ prompt, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0
52
  ):
53
  temperature = float(temperature)
54
  if temperature < 1e-2:
 
64
  seed=42,
65
  )
66
 
67
+ return ai.text_generation(prompt, **generate_kwargs, stream=False, details=False, return_full_text=True)
68
 
69
  @bot.event
70
  async def on_ready():