Update app.py

app.py

@@ -1,5 +1,5 @@
 from huggingface_hub import login
-from transformers import AutoTokenizer
+from transformers import AutoModelForCausalLM, AutoTokenizer
 from adapters import AutoAdapterModel
 import os
 import gradio as gr
@@ -11,10 +11,7 @@ login(token=HF_TOKEN)
 title = "Mental Health Chatbot"
 description = "This bot is using a fine-tuned version of meta-llama/Llama-2-7b-chat-hf"
 
-tokenizer = AutoTokenizer.from_pretrained("vojay/Llama-2-7b-chat-hf-mental-health")
-model = AutoAdapterModel.from_pretrained("meta-llama/Llama-2-7b-chat-hf", torch_dtype=torch.float16)
-adapter = model.load_adapter("vojay/Llama-2-7b-chat-hf-mental-health")
-model.active_adapters = adapter
+model = AutoModelForCausalLM.from_pretrained("vojay/Llama-2-7b-chat-hf-mental-health", torch_dtype=torch.float16)
 
 
 def predict(input, history=[]):