Update README.md
Browse files
README.md
CHANGED
@@ -52,7 +52,7 @@ X_train_emb = embedder.encode(X_train.tolist(), convert_to_numpy=True, show_prog
|
|
52 |
X_test_emb = embedder.encode(X_test.tolist(), convert_to_numpy=True, show_progress_bar=True)
|
53 |
|
54 |
# Load the trained classifier
|
55 |
-
model_path = hf_hub_download(repo_id="
|
56 |
model = joblib.load(model_path)
|
57 |
|
58 |
# Predict
|
|
|
52 |
X_test_emb = embedder.encode(X_test.tolist(), convert_to_numpy=True, show_progress_bar=True)
|
53 |
|
54 |
# Load the trained classifier
|
55 |
+
model_path = hf_hub_download(repo_id="selfconstruct3d/cybersec-classifier", filename="cybersec_classifier.pkl")
|
56 |
model = joblib.load(model_path)
|
57 |
|
58 |
# Predict
|