inference.py

import joblib
import numpy as np

model = joblib.load("model.pkl")

# Example input (5 features)
sample = np.array([[0.5, 0.2, 0.1, 0.7, 0.3]])

pred = model.predict(sample)

if pred[0] == 1:
    print("Malicious")
else:
    print("Benign")