Update README.md

README.md

@@ -62,7 +62,7 @@ class TextGenerationAssistant:
     def generate_response(self, user_query):
         prompt = self.format_prompt(user_query)
         outputs = self.pipe(prompt, **self.sampling_params)
-        return outputs[0]["generated_text"].split("[/INST]")[-1]
+        return outputs[0]["generated_text"].split("[/INST]")[1].strip()
 
 
 assistant = TextGenerationAssistant(model_id="Commencis/Commencis-LLM")
@@ -94,7 +94,7 @@ pipeline = transformers.pipeline(
 )
 
 outputs = pipeline(prompt, max_new_tokens=1024, do_sample=True, temperature=0.5, top_k=50, top_p=0.9)
-print(outputs[0]["generated_text"])
+print (outputs[0]["generated_text"].split("[/INST]")[1].strip())
 ```
 
 ## Bias, Risks, and Limitations