更新检测模型

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):
                 "<em class=\"color-gold\" id='score_{}'>".format(str(i)) + content_list[i] + "。\n" + "</em>")
         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)