Spaces:

VictorM-Coder
/

AIDetector

Running

VictorM-Coder commited on Feb 22

Commit

2195783

verified ·

1 Parent(s): 785a5d2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -25,7 +25,7 @@ from safetensors.torch import load_file
 # -----------------------------
 # MODEL INITIALIZATION
 # -----------------------------
-MODEL_NAME = "desklib/ai-text-detector-v1.01"
 tokenizer = None
 model = None
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

 # -----------------------------
 # MODEL INITIALIZATION
 # -----------------------------
+MODEL_NAME = "distilbert-base-uncased-finetuned-sst-2-english"
 tokenizer = None
 model = None
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")