smartbank-a25-cs078

pip install scikit-learn xgboost pandas joblib huggingface_hub

from huggingface_hub import hf_hub_download
import joblib
import pandas as pd


def load_model_from_hf(model_filename: str):
    model_path = hf_hub_download(
        repo_id="A25-CS078/smartbank-a25-cs078",
        filename=model_filename,
        cache_dir="models"
    )
    model = joblib.load(model_path)
    return model  # bisa disertai metadata jika diperlukan


def predict(model, data: dict):
    df = pd.DataFrame([data])

    prediction = model.predict(df)[0]

    proba = None
    if hasattr(model, "predict_proba"):
        proba = model.predict_proba(df)[0].tolist()

    return prediction, proba


# Load model
model = load_model_from_hf("model.pkl")

# Contoh input data
input_data = {
    "age": 35,
    "campaign": 1,
    "duration": 120,
    "euribor3m": 4.6
}

# Prediksi
prediction, probabilities = predict(model, input_data)