vojay commited on
Commit
295c595
·
verified ·
1 Parent(s): d4c17e3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -11,7 +11,8 @@ login(token=HF_TOKEN)
11
  title = "Mental Health Chatbot"
12
  description = "This bot is using a fine-tuned version of meta-llama/Llama-2-7b-chat-hf"
13
 
14
- model = AutoModelForCausalLM.from_pretrained("vojay/Llama-2-7b-chat-hf-mental-health", torch_dtype=torch.float16)
 
15
 
16
 
17
  def predict(input, history=[]):
 
11
  title = "Mental Health Chatbot"
12
  description = "This bot is using a fine-tuned version of meta-llama/Llama-2-7b-chat-hf"
13
 
14
+ model = AutoModelForCausalLM.from_pretrained(model_id)
15
+ model.load_adapter(adapter_model_id)
16
 
17
 
18
  def predict(input, history=[]):