Update g_retriever_pipeline.py

g_retriever_pipeline.py

@@ -11,6 +11,7 @@ class GRetrieverPipeline(Pipeline):
         self.max_txt_len = self.model.config.max_txt_len
         self.bos_length = len(self.model.config.bos_id)
         self.input_length = 0
+        self.prompt = "Generate a detailed review of a resume in relation to the current job market, presented as a textual graph. The review should be divided into three sections: strengths, weaknesses, and improvements."
         
     def _sanitize_parameters(self, **kwargs):
         preprocess_kwargs = {}
@@ -27,6 +28,7 @@ class GRetrieverPipeline(Pipeline):
 
     def preprocess(self, inputs, textualized_graph, graph, generate_kwargs=None):
         textualized_graph_ids = self.tokenizer(textualized_graph, add_special_tokens=False)["input_ids"][:self.max_txt_len]
+        prompt_ids = self.tokenizer(self.prompt, add_special_tokens=False)["input_ids"]
         question_ids = self.tokenizer(inputs, add_special_tokens=False)["input_ids"]
         eos_user_ids = self.tokenizer(self.eos_user, add_special_tokens=False)["input_ids"]
         
@@ -34,6 +36,7 @@ class GRetrieverPipeline(Pipeline):
             [-1]*(self.bos_length + 1)
             + textualized_graph_ids
             + question_ids
+            + prompt_ids
             + eos_user_ids
         ])
         model_inputs = {