Update pipeline.py

pipeline.py

@@ -40,9 +40,18 @@ class PreTrainedPipeline():
         questions, qg_examples = self.prepare_and_generate_questions(sents, answers)
         output = [{'answer': example['answer'], 'question': que} for example, que in zip(qg_examples, questions)]
         output = self.clean_generated_QAs(output, max_words_per_answer)
-        return [{"hello": "hi"}]
+        new = self.generate_probability_acceptances(inputs, output)
+        return [{"hello": new}]
         return output  
     
+    def generate_probability_acceptances(inputs, output):
+        model_inputs = []
+        for output in outputs:
+            text = f"Question: " + output['question'] + f" {split_char} Answer: " + output['answer'] + f" {split_char} Context: " + inputs
+            model_inputs.append(text)
+        return self.pipe(model_inputs)
+        
+    
     def prepare_and_generate_questions(self, sents, answers):
         qg_examples = self._prepare_inputs_for_qg_from_answers_hl(sents, answers)