diff --git a/chatgpt_detector_model_predict.py b/chatgpt_detector_model_predict.py index 66789f6..533e686 100644 --- a/chatgpt_detector_model_predict.py +++ b/chatgpt_detector_model_predict.py @@ -126,14 +126,14 @@ def main(content_list: list): reference_bool = is_reference_sentence(content_list[i]) if reference_bool == False: - if res["robot"] > 0.8: + if res["robot"] > 0.9: for _ in range(len(content_list[i])): gpt_score_list.append(res["robot"]) gpt_score_sentence_list.append(res["robot"]) sim_word += len(content_list[i]) gpt_content.append( "".format(str(i)) + content_list[i] + "\n" + "") - elif 0.8 >= res["robot"] > 0.3: + elif 0.9 >= res["robot"] > 0.5: for _ in range(len(content_list[i])): gpt_score_list.append(res["robot"]) gpt_score_sentence_list.append(res["robot"])