Spaces:

VictorM-Coder
/

AIDetector

Running

VictorM-Coder commited on Feb 22

Commit

eea664e

verified ·

1 Parent(s): 3b317ce

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
 # -----------------------------
 # 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 = "desklib/ai-text-detector-v1.03"
 tokenizer = None
 model = None
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")