cedpsam commited on
Commit
268e262
1 Parent(s): 70456c9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -17,8 +17,8 @@ def chat(message, history):
17
  bot_input_ids, max_length=80,
18
  pad_token_id=tokenizer.eos_token_id,
19
  top_p=0.92, top_k = 50,
20
- num_beams =8,
21
- max_time=20
22
  )
23
  response=tokenizer.decode(chat_history_ids[:, bot_input_ids.shape[-1]:][0], skip_special_tokens=True)
24
  history.append((message, response))
 
17
  bot_input_ids, max_length=80,
18
  pad_token_id=tokenizer.eos_token_id,
19
  top_p=0.92, top_k = 50,
20
+ num_beams =16,
21
+ max_time=30
22
  )
23
  response=tokenizer.decode(chat_history_ids[:, bot_input_ids.shape[-1]:][0], skip_special_tokens=True)
24
  history.append((message, response))