shakii commited on
Commit
4fa25e4
·
verified ·
1 Parent(s): 8c3f821

Update app.py

Browse files

Updated ai model to SuperAnnotate/ai-detector

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ app.add_middleware(
17
 
18
  # Load model and tokenizer
19
  #model_name = "fakespot-ai/roberta-base-ai-text-detection-v1"
20
- model_name = "desklib/ai-text-detector-v1.01"
21
  tokenizer = AutoTokenizer.from_pretrained(model_name)
22
  model = AutoModelForSequenceClassification.from_pretrained(model_name)
23
 
 
17
 
18
  # Load model and tokenizer
19
  #model_name = "fakespot-ai/roberta-base-ai-text-detection-v1"
20
+ model_name = "SuperAnnotate/ai-detector"
21
  tokenizer = AutoTokenizer.from_pretrained(model_name)
22
  model = AutoModelForSequenceClassification.from_pretrained(model_name)
23