Spaces:
Sleeping
Sleeping
EbubeJohnEnyi
commited on
Commit
•
8445ccb
1
Parent(s):
9aef249
Update app.py
Browse files
app.py
CHANGED
@@ -22,7 +22,8 @@ def generate_gpt2_response(question):
|
|
22 |
no_repeat_ngram_size=2,
|
23 |
top_k=10,
|
24 |
top_p=1,
|
25 |
-
temperature=0.9
|
|
|
26 |
generated_response = tokenizer.decode(generated_output[0], skip_special_tokens=True)
|
27 |
return generated_response
|
28 |
|
|
|
22 |
no_repeat_ngram_size=2,
|
23 |
top_k=10,
|
24 |
top_p=1,
|
25 |
+
temperature=0.9,
|
26 |
+
pad_token_id=model.config.pad_token_id)
|
27 |
generated_response = tokenizer.decode(generated_output[0], skip_special_tokens=True)
|
28 |
return generated_response
|
29 |
|