Spaces:

vojay
/

mental-health-demo

Runtime error

vojay commited on May 15, 2024

Commit

6a0a052

verified ·

1 Parent(s): 9a14ed3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,7 @@ adapter_model_id = "vojay/Llama-2-7b-chat-hf-mental-health"
 model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16)
 model.load_adapter(adapter_model_id)
-tokenizer = AutoTokenizer.from_pretrained(base_model)
 def get_base_prompt():

 model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16)
 model.load_adapter(adapter_model_id)
+tokenizer = AutoTokenizer.from_pretrained(model_id)
 def get_base_prompt():