jaelin215 commited on
Commit
2c17cd5
1 Parent(s): 69a01d3

updated path

Browse files
Files changed (1) hide show
  1. xgb_mental_health.py +2 -2
xgb_mental_health.py CHANGED
@@ -67,8 +67,8 @@ class MentalHealthClassifier:
67
 
68
  if __name__ == "__main__":
69
  tokenizer_model_name = "nlptown/bert-base-multilingual-uncased-sentiment"
70
- data_path = 'data/processed/data.csv'
71
- model_path = 'app/mental_health_model.pkl'
72
  mental_classifier = MentalHealthClassifier(data_path, model_path)
73
 
74
  if not os.path.exists(model_path):
 
67
 
68
  if __name__ == "__main__":
69
  tokenizer_model_name = "nlptown/bert-base-multilingual-uncased-sentiment"
70
+ data_path = 'data/data.csv'
71
+ model_path = 'mental_health_model.pkl'
72
  mental_classifier = MentalHealthClassifier(data_path, model_path)
73
 
74
  if not os.path.exists(model_path):