Hugging Face's logo

Spaces:
polygraf-ai
/
copyright_checker
Sleeping

App Files Community
aliasgerovs commited on
Commit
c6e6c67
·
1 Parent(s): fa7c450

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -180,7 +180,7 @@ AI DETECTION SECTION
180
  """
181
  device = "cuda" if torch.cuda.is_available() else "cpu"
182
 
183
- text_bc_model_path = "polygraf-ai/v3-bert-3-2m-trun-bc"
184
  text_bc_tokenizer = AutoTokenizer.from_pretrained(text_bc_model_path)
185
  text_bc_model = AutoModelForSequenceClassification.from_pretrained(text_bc_model_path).to(device)
186
 
 
180
  """
181
  device = "cuda" if torch.cuda.is_available() else "cpu"
182
 
183
+ text_bc_model_path = "polygraf-ai/v3-bert-3-2m-trun-bc-lighter-spec"
184
  text_bc_tokenizer = AutoTokenizer.from_pretrained(text_bc_model_path)
185
  text_bc_model = AutoModelForSequenceClassification.from_pretrained(text_bc_model_path).to(device)
186