Update pipeline.py
Browse files- pipeline.py +1 -0
pipeline.py
CHANGED
@@ -40,6 +40,7 @@ class PreTrainedPipeline():
|
|
40 |
questions, qg_examples = self.prepare_and_generate_questions(sents, answers)
|
41 |
output = [{'answer': example['answer'], 'question': que} for example, que in zip(qg_examples, questions)]
|
42 |
output = self.clean_generated_QAs(output, max_words_per_answer)
|
|
|
43 |
return output
|
44 |
|
45 |
def prepare_and_generate_questions(self, sents, answers):
|
|
|
40 |
questions, qg_examples = self.prepare_and_generate_questions(sents, answers)
|
41 |
output = [{'answer': example['answer'], 'question': que} for example, que in zip(qg_examples, questions)]
|
42 |
output = self.clean_generated_QAs(output, max_words_per_answer)
|
43 |
+
return [{"hello": "hi"}]
|
44 |
return output
|
45 |
|
46 |
def prepare_and_generate_questions(self, sents, answers):
|