Update app.py
Browse files
app.py
CHANGED
@@ -23,7 +23,7 @@ def tokenize_data(text):
|
|
23 |
|
24 |
def generate_answers(text):
|
25 |
inputs = tokenize_data(text)
|
26 |
-
results= model.generate(input_ids= inputs['input_ids']
|
27 |
max_length=64,
|
28 |
top_k=120,
|
29 |
top_p=0.98,
|
|
|
23 |
|
24 |
def generate_answers(text):
|
25 |
inputs = tokenize_data(text)
|
26 |
+
results= model.generate(input_ids= inputs['input_ids'], attention_mask=inputs['attention_mask'], do_sample=True,
|
27 |
max_length=64,
|
28 |
top_k=120,
|
29 |
top_p=0.98,
|