Update app.py

app.py

@@ -9,7 +9,6 @@ from custom_llm import CustomLLM, custom_chain_with_history
 
 API_TOKEN = os.getenv('HF_INFER_API')
 
-#API_URL = "https://api-inference.huggingface.co/models/gpt2"
 
 from typing import Optional
 
@@ -48,10 +47,10 @@ if prompt := st.chat_input("Ask me anything.."):
     response = st.session_state.chain.invoke(prompt).split("\n<|")[0]
 
     # Display assistant response in chat message container
-    with st.chat_message("Jojo"):
+    with st.chat_message("assistant"):
         st.markdown(response)
     st.session_state.memory.save_context({"question":prompt}, {"output":prompt})
     st.session_state.memory.chat_memory.messages = st.session_state.memory.chat_memory.messages[-15:]
     # Add assistant response to chat history
-    st.session_state.messages.append({"role": "Jojo", "content": response})
+    st.session_state.messages.append({"role": "assistant", "content": response})