Spaces:
Paused
Paused
Update UofTearsBot.py
Browse files- 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
|