aliasgerovs commited on
Commit
5aa28ef
·
1 Parent(s): 1303c4b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -175,7 +175,7 @@ AI DETECTION SECTION
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
 
 
175
  """
176
  device = "cuda" if torch.cuda.is_available() else "cpu"
177
 
178
+ text_bc_model_path = "polygraf-ai/v3-bert-3-2m-trun-bc"
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