Update chatbot.py

chatbot.py

@@ -1,20 +1,17 @@
 import os
-#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
-from langchain_community.llms import HuggingFaceHub
-from transformers import pipeline
 
 my_model_id = os.getenv('MODEL_REPO_ID', 'Default Value')
 token = os.getenv('HUGGINGFACEHUB_API_TOKEN')
 
 @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")
+    tokenizer = AutoTokenizer.from_pretrained(my_model_id)
+    model = AutoModelForCausalLM.from_pretrained(my_model_id)
     
     return tokenizer,model