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))
|