From 3b069e746c7f658b603f22aaa99635f7e23e74d4 Mon Sep 17 00:00:00 2001 From: "majiahui@haimaqingfan.com" Date: Tue, 16 Jul 2024 17:31:39 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=E6=A3=80=E6=B5=8B=E6=A8=A1?= =?UTF-8?q?=E5=9E=8B?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- chatgpt_detector_model_predict.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/chatgpt_detector_model_predict.py b/chatgpt_detector_model_predict.py index 0242104..c3a3f05 100644 --- a/chatgpt_detector_model_predict.py +++ b/chatgpt_detector_model_predict.py @@ -29,7 +29,7 @@ db_key_query = 'query' db_key_querying = 'querying' db_key_queryset = 'queryset' batch_size = 32 -model_name = "AIGC_detector_zhv2" +model_name = "drop_aigc_model_2" tokenizer = AutoTokenizer.from_pretrained(model_name) model = AutoModelForSequenceClassification.from_pretrained(model_name).cpu() @@ -101,7 +101,7 @@ def main(content_list: list): "".format(str(i)) + content_list[i] + "。\n" + "") else: gpt_score_list.append(0) - gpt_content.append(content_list[i] + "。\n") + gpt_content.append(content_list[i] + "\n") return_list["gpt_content"] = "".join(gpt_content) return_list["gpt_score_list"] = str(gpt_score_list)