Update app.py

app.py

@@ -154,12 +154,12 @@ def thread_response(msg_history: str, msg_prompt: str) -> dict:
     
     generation_params = {"do_sample": True,"temperature": 0.7,"top_p": 0.95,"top_k": 40,"max_new_tokens": 512,"repetition_penalty": 1.1}
     pipe = pipeline("text-generation",model=model, tokenizer=tokenizer, **generation_params)
-    message_history= f''' incase you are asked about something you talked about the previous days or times or periods,  use this as reference to recall {msg_history}. Do not point out everything just major keypoints. If nothing is mentioned about the past conversation ingore this. '''
+    instructions= f''' incase you are asked about something you talked about the previous days or times or periods,  use this as reference to recall {msg_history}. Do not point out everything just major keypoints. If nothing is mentioned about the past conversation ingore this. '''
 
-    if previous_chat:
+    if msg_history:
         
         prompt_template=f'''
-        <|im_start|>system{system_message}{message_history}<|im_end|>
+        <|im_start|>system{system_message}{instructions}<|im_end|>
         <|im_start|>user {msg_prompt}<|im_end|>
         <|im_start|>assistant
         '''
@@ -202,12 +202,12 @@ def chat_thread_response(ai_persona: str,msg_history: str, msg_prompt: str) -> d
     
     generation_params = {"do_sample": True,"temperature": 0.7,"top_p": 0.95,"top_k": 40,"max_new_tokens": 512,"repetition_penalty": 1.1}
     pipe = pipeline("text-generation",model=model, tokenizer=tokenizer, **generation_params)
-    message_history= f''' incase you are asked about something you talked about the previous days or times or periods,  use this as reference to recall {msg_history}. Do not point out everything just major keypoints. If nothing is mentioned about the past conversation ingore this. '''
+    instructions= f''' Incase you are asked about something you talked about the previous days or times or periods,  use this as reference to recall {msg_history}. Do not point out everything just major keypoints. If nothing is mentioned about the past conversation ingore this. '''
     
-    if previous_chat:
+    if msg_history:
         
         prompt_template=f'''
-        <|im_start|>system{ai_persona}{message_history}<|im_end|>
+        <|im_start|>system{ai_persona}{instructions}<|im_end|>
         <|im_start|>user {msg_prompt}<|im_end|>
         <|im_start|>assistant
         '''