KomalNaseem commited on
Commit
82679c0
·
verified ·
1 Parent(s): 6252f8b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -3,6 +3,8 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
3
  import torch
4
  import re
5
 
 
 
6
  model_path = "KomalNaseem/depression_model_part1"
7
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
8
 
 
3
  import torch
4
  import re
5
 
6
+ login(token=os.getenv("HF_TOKEN"))
7
+
8
  model_path = "KomalNaseem/depression_model_part1"
9
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
10