From 05ed9b00104b3da2e9cb0d541b5c5bbceb027fde Mon Sep 17 00:00:00 2001 From: Christian Krinitsin Date: Sun, 1 Jun 2025 21:19:28 +0200 Subject: adjust classifier --- classification/output.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'classification/output.py') diff --git a/classification/output.py b/classification/output.py index df64dbcc..971c3830 100644 --- a/classification/output.py +++ b/classification/output.py @@ -1,7 +1,7 @@ from os import path, makedirs -def output(text : str, labels : list, scores : list, identifier : str): - file_path = f"output/{labels[0]}/{identifier}" +def output(text : str, category : str, labels : list, scores : list, identifier : str): + file_path = f"output/{category}/{identifier}" makedirs(path.dirname(file_path), exist_ok = True) with open(file_path, "w") as file: -- cgit 1.4.1