Bendo commited on
Commit
ab3cc0a
1 Parent(s): 8911053

Update question_gen.py

Browse files
Files changed (1) hide show
  1. question_gen.py +1 -1
question_gen.py CHANGED
@@ -21,5 +21,5 @@ 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 = "\n".join([x.replace("question:", "") for x in dec])
25
  return Question
21
 
22
  dec = [tknizer.decode(ids,skip_special_tokens=True) for ids in outs]
23
 
24
+ Question = [x.replace("question:", "") for x in dec]
25
  return Question