Upload app.py

app.py

@@ -5,9 +5,17 @@ from model import DepressionClassifier
 import hopsworks
 import joblib
 import torch
+from huggingface_hub import hf_hub_download
+
 class_names = ['Not Depressed', 'Depressed']
+pt_file = hf_hub_download(repo_id="liangc40/sentimental_analysis", filename="model.pt")
+
 model = DepressionClassifier(len(class_names), 'bert-base-cased')
-model.load_state_dict(torch.load("liangc40/sentimental_analysis/model.pt"))
+model.load_state_dict(torch.load(pt_file))
+
+
+
+
 model.eval()
 #pipe = pipeline(model="liangc40/sentimental_analysis")