Update app.py

app.py

@@ -36,7 +36,7 @@ def generate_response(input_text):
         system_prompt = get_system_prompt()
         st.session_state['conversation'].append(f"System: {system_prompt}")
         system_input_ids = tokenizer.encode(system_prompt + tokenizer.eos_token, return_tensors='pt').to(device)
-        st.session_state['history'] = system_input_ids[0].tolist()
+        st.session_state['history'] = system_input_ids[0].tolist()  # Save only the system prompt to history
 
     # Encode the new user input, add end of string token
     new_user_input_ids = tokenizer.encode(input_text + tokenizer.eos_token, return_tensors='pt').to(device)