Spaces:
Sleeping
Sleeping
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 |
|