Update model.py

model.py

@@ -6,6 +6,7 @@ from langchain.chains import LLMChain, ConversationChain
 from langchain_community.llms.huggingface_pipeline import HuggingFacePipeline
 from langchain_community.llms import HuggingFaceEndpoint
 from langchain.prompts import PromptTemplate, ChatPromptTemplate
+from langchain_core.prompts.chat import SystemMessagePromptTemplate, HumanMessagePromptTemplate
 import streamlit as st
 
 your_endpoint_url = "https://kp4xdy196cw81uf3.us-east-1.aws.endpoints.huggingface.cloud"