Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -51,8 +51,8 @@ def chat_with_distilgpt2(input_text):
|
|
51 |
no_repeat_ngram_size=3,
|
52 |
repetition_penalty=1.2,
|
53 |
temperature=0.7,
|
54 |
-
top_k=
|
55 |
-
top_p=0.
|
56 |
early_stopping=True,
|
57 |
pad_token_id=tokenizer.eos_token_id,
|
58 |
eos_token_id=tokenizer.eos_token_id
|
|
|
51 |
no_repeat_ngram_size=3,
|
52 |
repetition_penalty=1.2,
|
53 |
temperature=0.7,
|
54 |
+
top_k=20,
|
55 |
+
top_p=0.8,
|
56 |
early_stopping=True,
|
57 |
pad_token_id=tokenizer.eos_token_id,
|
58 |
eos_token_id=tokenizer.eos_token_id
|