Spaces:

yagnik12
/

AI_Text_Detecter

Running

yagnik12 commited on Sep 12

Commit

6ffc9d2

verified ·

1 Parent(s): 945bad0

Update ai_text_detector_valid_final.py

Files changed (1) hide show

ai_text_detector_valid_final.py CHANGED Viewed

@@ -7,7 +7,9 @@ MODELS = {
     "DeBERTa Detector": "distilbert-base-uncased-finetuned-sst-2-english",
     "MonkeyDAnh":"MonkeyDAnh/my-awesome-ai-detector-roberta-base-v4-human-vs-machine-finetune",
     "Andreas122001":"andreas122001/roberta-academic-detector",
-    "Microsoft":"microsoft/deberta-v3-small-mnli"
 }
 def load_model(model_id):

     "DeBERTa Detector": "distilbert-base-uncased-finetuned-sst-2-english",
     "MonkeyDAnh":"MonkeyDAnh/my-awesome-ai-detector-roberta-base-v4-human-vs-machine-finetune",
     "Andreas122001":"andreas122001/roberta-academic-detector",
+    "mrm8488":"mrm8488/deberta-v3-small-finetuned-mnli",
+      "rdp99":"rdp99/deberta-v3-small-finetuned-mnli",
+      "cross_encoder":"cross-encoder/nli-deberta-v3-small"
 }
 def load_model(model_id):