p-christ commited on
Commit
4abe38e
1 Parent(s): ea92555

Update pipeline.py

Browse files
Files changed (1) hide show
  1. 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):