Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -12,7 +12,11 @@ HF_TOKEN = os.getenv('mentalhealth_llama_chat')
|
|
12 |
login(HF_TOKEN)
|
13 |
|
14 |
# Model name in Hugging Face docs
|
15 |
-
model = '
|
|
|
|
|
|
|
|
|
16 |
|
17 |
|
18 |
tokenizer = AutoTokenizer.from_pretrained(model, use_auth_token=True)
|
|
|
12 |
login(HF_TOKEN)
|
13 |
|
14 |
# Model name in Hugging Face docs
|
15 |
+
model = 'vibhorag101/llama-2-7b-chat-hf-phr_mental_therapy'
|
16 |
+
|
17 |
+
|
18 |
+
#other model that we were using before
|
19 |
+
#'klyang/MentaLLaMA-chat-7B'
|
20 |
|
21 |
|
22 |
tokenizer = AutoTokenizer.from_pretrained(model, use_auth_token=True)
|