Updated

app.py

@@ -22,16 +22,16 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 print(f"Using device: {device}")
 
 models = {
-    "Polygraf AI Watson (Base Model)": AutoModelForSequenceClassification.from_pretrained(
+    "Polygraf AI (Base Model)": AutoModelForSequenceClassification.from_pretrained(
         "polygraf-ai/bc-roberta-openai-2sent"
     ).to(device),
-    "Polygraf AI Sherlock (Advanced Model)": AutoModelForSequenceClassification.from_pretrained(
+    "Polygraf AI (Advanced Model)": AutoModelForSequenceClassification.from_pretrained(
         "polygraf-ai/bc_combined_3sent"
     ).to(device),
 }
 tokenizers = {
-    "Polygraf AI Watson (Base Model)": AutoTokenizer.from_pretrained("polygraf-ai/bc-roberta-openai-2sent"),
-    "Polygraf AI Sherlock (Advanced Model)": AutoTokenizer.from_pretrained("polygraf-ai/bc_combined_3sent"),
+    "Polygraf AI (Base Model)": AutoTokenizer.from_pretrained("polygraf-ai/bc-roberta-openai-2sent"),
+    "Polygraf AI (Advanced Model)": AutoTokenizer.from_pretrained("polygraf-ai/bc_combined_3sent"),
 }
 
 
@@ -151,8 +151,8 @@ def process_text(text, model="BC Original"):
 
 
 ai_check_options = [
-    "Polygraf AI Watson (Base Model)",
-    "Polygraf AI Sherlock (Advanced Model)",
+    "Polygraf AI (Base Model)",
+    "Polygraf AI (Advanced Model)",
 ]
 
 
@@ -202,7 +202,7 @@ def ai_generated_test_gptzero(text):
     return result, None
 
 
-def highlighter_polygraf(text, model="Polygraf AI Watson (Base Model)"):
+def highlighter_polygraf(text, model="Polygraf AI (Base Model)"):
     return process_text(text=text, model=model)