Avinash250325 commited on
Commit
e15a615
·
verified ·
1 Parent(s): eba6de6

Update backend.py

Browse files
Files changed (1) hide show
  1. 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