Spaces:
Runtime error
Runtime error
Update question_gen.py
Browse files- 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 = "
|
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
|