new commits

celebbot.py

@@ -171,7 +171,7 @@ class CelebBot():
             else:
               knowledge = self.retrieve_knowledge_assertions()
 
-            query = f"Context: {instruction} {knowledge}\n\nChat History: {chat_his}Question: {self.text}\n\nAnswer:"
+            query = f"Context: {instruction} {knowledge}\n\nQuestion: {self.text}\n\nAnswer:"
             input_ids = self.QA_tokenizer(f"{query}", truncation=False, return_tensors="pt").input_ids.to(self.QA_model.device)
             outputs = self.QA_model.generate(input_ids, max_length=1024, min_length=8, repetition_penalty=2.5)
             self.text = self.QA_tokenizer.decode(outputs[0], skip_special_tokens=True)