Update README.md

README.md

@@ -61,21 +61,27 @@ A **fast, accurate Small Language Model (SLM)** that detects secrets in code bef
 ```python
 from transformers import pipeline
 
+# Load the pipeline with the secrets detection model
 classifier = pipeline(
     "text-classification",
     model="hypn05/secrets-sentinel"
 )
 
-# Examples
-results = classifier([
-    "password='supersecret123'",           # ✓ DETECTS SECRET
-    "api_key = 'sk-1234567890abc'",        # ✓ DETECTS SECRET
-    "print('Hello, world!')",              # ✓ Normal code
-    "def calculate_sum(a, b):",            # ✓ Normal code
-])
-
-for text, result in zip(["Test1", "Test2", "Test3", "Test4"], results):
-    print(f"{result['label']:>8} | {text}")
+# Define the input examples
+inputs = [
+    "password='supersecret123'",           # Expected: Secret
+    "api_key = 'sk-1234567890abc'",        # Expected: Secret
+    "print('Hello, world!')",              # Expected: Safe
+    "def calculate_sum(a, b):",            # Expected: Safe
+]
+
+# Run the classifier on the inputs
+results = classifier(inputs)
+
+# Print the actual input string and the result ("Secret" if LABEL_1, else "Safe")
+for input_text, result in zip(inputs, results):
+    label = "Secret" if result['label'] == "LABEL_1" else "Safe"
+    print(f"{label} | {input_text}")
 ```
 
 ## Advanced Usage