Update README.md

README.md

@@ -27,6 +27,31 @@ Trained on the [Xerv-AI/netuark-posts-6000](https://huggingface.co/datasets/Xerv
 ## Usage
 ```python
 import joblib
-model = joblib.load('netuark_ensemble_classifier.joblib')
-prediction = model.predict(["New AI breakthrough on HackerNews"])
+import os
+from huggingface_hub import hf_hub_download
+
+# Define the missing custom function required by the unpickler
+def advanced_clean(text):
+    return text
+
+# Assign it to __main__ to ensure joblib can find it during loading
+import __main__
+__main__.advanced_clean = advanced_clean
+
+# Repository and filename
+repo_id = 'Phase-Technologies/netuark-classifier-ensemble'
+filename = 'netuark_ensemble_classifier.joblib'
+
+try:
+    # Download the file from Hugging Face
+    file_path = hf_hub_download(repo_id=repo_id, filename=filename)
+
+    # Load the model
+    model = joblib.load(file_path)
+    prediction = model.predict(["📰 Perplexity's 'Personal Computer' Lets AI Agents Access Your Local Files #slashdot"])
+    print(f"Prediction: {prediction}")
+except Exception as e:
+    import traceback
+    print(f"An error occurred: {e}")
+    traceback.print_exc()
 ```