42Cummer commited on
Commit
bf06b09
·
verified ·
1 Parent(s): b48d8e8

Update UofTearsBot.py

Browse files
Files changed (1) hide show
  1. UofTearsBot.py +2 -2
UofTearsBot.py CHANGED
@@ -4,10 +4,10 @@ from IllnessClassifier import IllnessClassifier
4
  from typing import List, Dict
5
 
6
  class UofTearsBot(object):
7
- def __init__(self, threshold: float = 0.86, max_history_msgs: int = 50):
8
  self.suicidality_detector = SIDetector()
9
  self.illness_classifier = IllnessClassifier()
10
- self.chatbot = AdviceGenerator()
11
  self.history: List[Dict[str, str]] = []
12
  self.FLAG = False # suicidal crisis flag
13
  self.threshold = threshold
 
4
  from typing import List, Dict
5
 
6
  class UofTearsBot(object):
7
+ def __init__(self, llm, threshold: float = 0.86, max_history_msgs: int = 50):
8
  self.suicidality_detector = SIDetector()
9
  self.illness_classifier = IllnessClassifier()
10
+ self.chatbot = AdviceGenerator(llm)
11
  self.history: List[Dict[str, str]] = []
12
  self.FLAG = False # suicidal crisis flag
13
  self.threshold = threshold