Spaces:

qisan
/

Depressed_sentimental_analysis

Runtime error

qisan commited on Jan 13, 2023

Commit

358197a

•

1 Parent(s): e654eff

Upload app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ 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(pt_file), map_location=torch.device('cpu'))

 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(pt_file, map_location=torch.device('cpu')))