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

Update app.py

Browse files

Updated model to desklib/ai-text-detector-v1.01

Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -16,7 +16,8 @@ app.add_middleware(
16
  )
17
 
18
  # Load model and tokenizer
19
- model_name = "fakespot-ai/roberta-base-ai-text-detection-v1"
 
20
  tokenizer = AutoTokenizer.from_pretrained(model_name)
21
  model = AutoModelForSequenceClassification.from_pretrained(model_name)
22
 
 
16
  )
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