Updated

app.py

@@ -186,7 +186,7 @@ text_bc_model_path = "polygraf-ai/v3-bert-3-2m-trun-bc-lighter-spec"
 text_bc_tokenizer = AutoTokenizer.from_pretrained(text_bc_model_path)
 text_bc_model = AutoModelForSequenceClassification.from_pretrained(text_bc_model_path).to(device)
 
-text_mc_model_path = "polygraf-ai/text-detect-mc-bert-base-uncased-v1-bert-429k-256"
+text_mc_model_path = "polygraf-ai/text-detect-mc-bert-base-uncased-v1-bert-429k-lighter-spec"
 text_mc_tokenizer = AutoTokenizer.from_pretrained(text_mc_model_path)
 text_mc_model = AutoModelForSequenceClassification.from_pretrained(text_mc_model_path).to(device)
 
@@ -287,7 +287,7 @@ def ai_generated_test(ai_option, input):
     
     for i in range(samples_len):
         cleaned_text_bc = remove_special_characters(segments[i])
-        cleaned_text_mc = remove_special_characters_2(segments[i])
+        cleaned_text_mc = remove_special_characters(segments[i])
         bc_score = predict_bc(text_bc_model, text_bc_tokenizer,cleaned_text_bc )
         mc_score = predict_mc(text_mc_model, text_mc_tokenizer, cleaned_text_mc)
         bc_scores.append(bc_score)