Merge pull request #257 from TogetherCrew/feat/218-violation-detection

feat: violation detection, using our previously fine tuned LLM!
TogetherCrew · Aug 12, 2024 · aaa5090 · aaa5090
2 parents 30e2f57 + 7fadad0
commit aaa5090
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dags/violation_detection_helpers/utils/classify.py b/dags/violation_detection_helpers/utils/classify.py
@@ -6,7 +6,7 @@ def __init__(self) -> None:
         self.client = OpenAI()
 
         # configurations
-        self.model = "gpt-3.5-turbo"
+        self.model = "ft:gpt-3.5-turbo-0125:togethercrew::9mHnDhmo"
         self.system_content = "You are a classifier with a PhD in psychology and 30 years of experience teaching non-violent communication. "
         self.user_context = (
             "[INSTRUCTION]\n"