Update backend.py
Browse files- backend.py +2 -2
backend.py
CHANGED
|
@@ -40,10 +40,10 @@ def get_question(tag, difficulty, context, answer="", num_questions=1, max_lengt
|
|
| 40 |
input_ids=features['input_ids'],
|
| 41 |
attention_mask=features['attention_mask'],
|
| 42 |
max_length=max_length,
|
|
|
|
| 43 |
do_sample=True,
|
| 44 |
top_p=0.95,
|
| 45 |
-
top_k=50
|
| 46 |
-
num_return_sequences=num_questions
|
| 47 |
)
|
| 48 |
|
| 49 |
# Decode questions
|
|
|
|
| 40 |
input_ids=features['input_ids'],
|
| 41 |
attention_mask=features['attention_mask'],
|
| 42 |
max_length=max_length,
|
| 43 |
+
num_return_sequences=num_questions,
|
| 44 |
do_sample=True,
|
| 45 |
top_p=0.95,
|
| 46 |
+
top_k=50
|
|
|
|
| 47 |
)
|
| 48 |
|
| 49 |
# Decode questions
|