Update chatbot.py

chatbot.py

@@ -13,17 +13,10 @@ 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("KvrParaskevi/Hotel-Assistant-Attempt4-Llama-2-7b")
+    model = AutoModelForCausalLM.from_pretrained("KvrParaskevi/Hotel-Assistant-Attempt4-Llama-2-7b")
     
-    # Replace 'username/model_name' with your model's identifier on the Hugging Face Model Hub
-    model_identifier = "KvrParaskevi/Hotel-Assistant-Attempt4-Llama-2-7b"
-    task = "text-classification"  # Change this to your model's task
-    
-    # Load the model using the pipeline
-    model_pipeline = pipeline(task, model=model_identifier)
-
-    return model_pipeline
+    return tokenizer,model
 
 def demo_miny_memory(model):
     # llm_data = get_Model(hugging_face_key)