Spaces:

shakii
/

textdetectextension

Sleeping

shakii commited on Mar 26

Commit

8990b6e

verified ·

1 Parent(s): 4fa25e4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,8 +16,8 @@ app.add_middleware(
 )
 # Load model and tokenizer
-#model_name = "fakespot-ai/roberta-base-ai-text-detection-v1"
-model_name = "SuperAnnotate/ai-detector"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSequenceClassification.from_pretrained(model_name)

 )
 # Load model and tokenizer
+model_name = "fakespot-ai/roberta-base-ai-text-detection-v1"
+#model_name = "SuperAnnotate/ai-detector"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSequenceClassification.from_pretrained(model_name)