Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 =
|
21 |
-
max_time=
|
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))
|