rishiraj commited on
Commit
5aa5960
1 Parent(s): 6dcf94d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ def format_prompt(message, history):
15
  return prompt
16
 
17
  def generate(prompt, history, email):
18
- generate_kwargs = dict(seed=42)
19
 
20
  formatted_prompt = format_prompt(f"{prompt}", history)
21
  stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
 
15
  return prompt
16
 
17
  def generate(prompt, history, email):
18
+ generate_kwargs = dict(max_new_tokens=256, seed=42)
19
 
20
  formatted_prompt = format_prompt(f"{prompt}", history)
21
  stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)