Update demo_label.py

BenderV · Nov 6, 2024 · 73d3da9 · 73d3da9
1 parent b3bc0a0
commit 73d3da9
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/examples/demo_label.py b/examples/demo_label.py
@@ -10,13 +10,8 @@ def label_item(category: str, from_response: Message):
     # TODO: Implement function
     raise NotImplementedError()
 
-
-with open("./function_label.json") as f:
-    FUNCTION_LABEL_ITEM = json.load(f)
-
-classifierGPT = Autochat.from_template("./classify_template.txt")
-
-classifierGPT.add_function(label_item, FUNCTION_LABEL_ITEM)
+classifierGPT = Autochat("you classify title")
+classifierGPT.add_function(label_item)
 
 text = "The new iPhone is out"
 for message in classifierGPT.run_conversation(text):