Update chatbot.py

chatbot.py

@@ -7,7 +7,7 @@ from langchain_community.llms.huggingface_pipeline import HuggingFacePipeline
 from transformers import AutoModelForCausalLM, AutoTokenizer,  BitsAndBytesConfig, pipeline
 #import streamlit as st
 
-my_model_id = "KvrParaskevi/Llama-2-7b-Hotel-Booking-Model" #os.getenv('MODEL_REPO_ID', 'Default Value')
+my_model_id = os.getenv('MODEL_REPO_ID', 'Default Value')
 token = os.getenv('HUGGINGFACEHUB_API_TOKEN')
 
 template = """<<SYS>>
@@ -64,11 +64,11 @@ def demo_miny_memory():
     memory = ConversationBufferMemory(memory_key="chat_history", llm = llm)
     return memory
 
-def demo_chain(input_text):
+def demo_chain(input_text, memory):
     conversation = ConversationChain(
         llm=llm,
         verbose=True,
-        memory=demo_miny_memory()
+        memory=memory
     )
 
     chat_reply = conversation(input=input_text)