Update app.py

app.py

@@ -175,7 +175,7 @@ AI DETECTION SECTION
 """
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-text_bc_model_path = "polygraf-ai/ai-text-bc-bert-2-7m"
+text_bc_model_path = "polygraf-ai/v3-bert-3-2m-trun-bc"
 text_bc_tokenizer = AutoTokenizer.from_pretrained(text_bc_model_path)
 text_bc_model = AutoModelForSequenceClassification.from_pretrained(text_bc_model_path).to(device)
 
@@ -192,8 +192,9 @@ def update_character_count(text):
 
 def predict_bc(model, tokenizer, text):
     tokens = tokenizer(
-        text, padding=True, truncation=True, return_tensors="pt"
+        text, padding='max_length', truncation=True, max_length=256, return_tensors="pt"
     ).to(device)["input_ids"]
+    
     output = model(tokens)
     output_norm = softmax(output.logits.detach().cpu().numpy(), 1)[0]
     print("BC Score: ", output_norm)