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