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
|