Spaces:

KomalNaseem
/

depression_chatbot

Sleeping

KomalNaseem commited on May 17

Commit

6252f8b

verified ·

1 Parent(s): 07a2e9c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 import re
-model_path = "./depression_model_part1"
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 tokenizer = AutoTokenizer.from_pretrained(model_path)

 import torch
 import re
+model_path = "KomalNaseem/depression_model_part1"
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 tokenizer = AutoTokenizer.from_pretrained(model_path)