aliasgerovs commited on
Commit
6aa7315
·
1 Parent(s): 5aa28ef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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/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
 
182
- text_mc_model_path = "polygraf-ai/ai-text-mc-v5-lighter-spec"
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