Update chatbot_bedrock.py

chatbot_bedrock.py

@@ -1,14 +1,16 @@
 import os
-from langchain import PromptTemplate, HuggingFaceHub, LLMChain
+#from langchain import PromptTemplate, HuggingFaceHub, LLMChain
 from langchain.memory import ConversationBufferMemory
 from langchain.chains import ConversationChain
 import langchain.globals
 from transformers import AutoModelForCausalLM, AutoTokenizer
+import streamlit as st
 
-def get_Model():
+@st.cache_resource 
+def load_model():
     tokenizer = AutoTokenizer.from_pretrained("KvrParaskevi/Hotel-Assistant-Attempt4-Llama-2-7b")
     model = AutoModelForCausalLM.from_pretrained("KvrParaskevi/Hotel-Assistant-Attempt4-Llama-2-7b")
-    return model
+    return tokenizer,model
 
 
 #Write function to connect to Bedrock