migueldeguzmandev
commited on
Commit
•
aea8084
1
Parent(s):
5d6c426
Update app.py
Browse files
app.py
CHANGED
@@ -22,8 +22,10 @@ def generate_response(input_text, temperature):
|
|
22 |
attention_mask=attention_mask,
|
23 |
max_length=300,
|
24 |
num_return_sequences=1,
|
25 |
-
no_repeat_ngram_size=2,
|
26 |
temperature=temperature,
|
|
|
|
|
|
|
27 |
do_sample=True, # Set do_sample to True when using temperature
|
28 |
)
|
29 |
|
|
|
22 |
attention_mask=attention_mask,
|
23 |
max_length=300,
|
24 |
num_return_sequences=1,
|
|
|
25 |
temperature=temperature,
|
26 |
+
no_repeat_ngram_size=2,
|
27 |
+
top_k=50,
|
28 |
+
top_p=0.95,
|
29 |
do_sample=True, # Set do_sample to True when using temperature
|
30 |
)
|
31 |
|