Spaces:
Running
Running
aliasgerovs
commited on
Commit
·
6aa7315
1
Parent(s):
5aa28ef
Update app.py
Browse files
app.py
CHANGED
@@ -175,11 +175,11 @@ AI DETECTION SECTION
|
|
175 |
"""
|
176 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
177 |
|
178 |
-
text_bc_model_path = "polygraf-ai/
|
179 |
text_bc_tokenizer = AutoTokenizer.from_pretrained(text_bc_model_path)
|
180 |
text_bc_model = AutoModelForSequenceClassification.from_pretrained(text_bc_model_path).to(device)
|
181 |
|
182 |
-
text_mc_model_path = "polygraf-ai/
|
183 |
text_mc_tokenizer = AutoTokenizer.from_pretrained(text_mc_model_path)
|
184 |
text_mc_model = AutoModelForSequenceClassification.from_pretrained(text_mc_model_path).to(device)
|
185 |
|
|
|
175 |
"""
|
176 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
177 |
|
178 |
+
text_bc_model_path = "polygraf-ai/ai-text-bc-bert-2-7m"
|
179 |
text_bc_tokenizer = AutoTokenizer.from_pretrained(text_bc_model_path)
|
180 |
text_bc_model = AutoModelForSequenceClassification.from_pretrained(text_bc_model_path).to(device)
|
181 |
|
182 |
+
text_mc_model_path = "polygraf-ai/text-detect-mc-bert-base-uncased-v1-bert-429k"
|
183 |
text_mc_tokenizer = AutoTokenizer.from_pretrained(text_mc_model_path)
|
184 |
text_mc_model = AutoModelForSequenceClassification.from_pretrained(text_mc_model_path).to(device)
|
185 |
|