Spaces:

Hassan175
/

suicide-detection

Sleeping

Hassan175 commited on Feb 8, 2023

Commit

01365ee

1 Parent(s): bc3ff5d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ import pickle
 # define constants from training
 MODEL_PATH = "./model.pickle"
 TOKENIZER_PATH = "./tokenizer.pickle"
-INDEX_CLASS = {1:'suicide', 0:'non-suicide'}
 MAX_SEQ_DF = 128

 # define constants from training
 MODEL_PATH = "./model.pickle"
 TOKENIZER_PATH = "./tokenizer.pickle"
+INDEX_CLASS = {1:'Suicidal', 0:'non-Suicidal'}
 MAX_SEQ_DF = 128