Spaces:

polygraf-ai
/

copyright_checker

Running

aliasgerovs commited on Feb 28

Commit

9545dbd

•

1 Parent(s): c2db75c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -184,7 +184,7 @@ AI DETECTION SECTION
 """
 device = "cuda" if torch.cuda.is_available() else "cpu"
-text_bc_model_path = "polygraf-ai/v3-bert-3-2m-trun-bc-lighter-spec"
 text_bc_tokenizer = AutoTokenizer.from_pretrained(text_bc_model_path)
 text_bc_model = AutoModelForSequenceClassification.from_pretrained(text_bc_model_path).to(device)

 """
 device = "cuda" if torch.cuda.is_available() else "cpu"
+text_bc_model_path = "polygraf-ai/text-detect-bc-v11-4m"
 text_bc_tokenizer = AutoTokenizer.from_pretrained(text_bc_model_path)
 text_bc_model = AutoModelForSequenceClassification.from_pretrained(text_bc_model_path).to(device)