Bendo commited on
Commit
4530f39
·
1 Parent(s): 985d405

Update question_gen.py

Browse files
Files changed (1) hide show
  1. question_gen.py +1 -1
question_gen.py CHANGED
@@ -21,6 +21,6 @@ def get_question(sentence,answer,mdl,tknizer, num_seq, num_beams, max_length):
21
 
22
  dec = [tknizer.decode(ids,skip_special_tokens=True) for ids in outs]
23
 
24
- Question = " ".join([x.replace("question:", "") for x in dec])
25
  Question = Question.strip()
26
  return Question
 
21
 
22
  dec = [tknizer.decode(ids,skip_special_tokens=True) for ids in outs]
23
 
24
+ Question = "\n".join([x.replace("question:", "") for x in dec])
25
  Question = Question.strip()
26
  return Question