Spaces:

bobpopboom
/

Mental_chat

Sleeping

bobpopboom commited on Feb 10

Commit

5827377

verified ·

1 Parent(s): 68cf9c5

maybe now

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,8 +5,9 @@ import torch
 # Determine device (ctranslate2 handles device placement internally)
 device = "cuda" if torch.cuda.is_available() else "cpu"  # Still useful for other ops
-model_path =  "mradermacher/TinyLlama-Friendly-Psychotherapist-GGUF/TinyLlama-Friendly-Psychotherapist.Q4_K_S.gguf"
 try:
     # 1. Load the tokenizer (same as before)

 # Determine device (ctranslate2 handles device placement internally)
 device = "cuda" if torch.cuda.is_available() else "cpu"  # Still useful for other ops
+model="thrishala/mental_health_chatbot"
+model_download_link =  "https://huggingface.co/mradermacher/TinyLlama-Friendly-Psychotherapist-GGUF/resolve/main/TinyLlama-Friendly-Psychotherapist.Q4_K_S.gguf"
+model_path = "TinyLlama-Friendly-Psychotherapist.Q4_K_S.gguf" # gguf
 try:
     # 1. Load the tokenizer (same as before)