16 lines
402 B
Python
16 lines
402 B
Python
import joblib
|
|
|
|
vectorizer = joblib.load("xss_vectorizer.joblib")
|
|
clf = joblib.load("xss_classifier.joblib")
|
|
|
|
print("Type text to classify. Ctrl+C or empty line to quit.\n")
|
|
while True:
|
|
text = input("> ").strip()
|
|
if not text:
|
|
break
|
|
X = vectorizer.transform([text])
|
|
pred = clf.predict(X)[0]
|
|
verdict = "MALICIOUS" if pred == 1 else "benign"
|
|
print(f"[{verdict}] {text}\n")
|
|
|