Views
No views yet
1pip install xgboost sentence-transformers huggingface_hub
2import xgboost as xgb
3from sentence_transformers import SentenceTransformer
4from huggingface_hub import hf_hub_download
5
6# -----------------------------
7# 1. Download model from Hugging Face Hub
8# -----------------------------
9REPO_ID = "mjpsm/Shujaa-xgb-model"
10FILENAME = "Shujaa_xgb_model.json"
11
12model_path = hf_hub_download(repo_id=REPO_ID, filename=FILENAME)
13
14# -----------------------------
15# 2. Load model + embedder
16# -----------------------------
17model = xgb.XGBRegressor()
18model.load_model(model_path)
19
20embedder = SentenceTransformer("all-mpnet-base-v2")
21
22# -----------------------------
23# 3. Example prediction
24# -----------------------------
25text = "I carried my responsibility with persistence until it was complete."
26embedding = embedder.encode([text])
27score = model.predict(embedding)[0]
28
29print("Predicted Shujaa Score:", round(float(score), 3))