Update README.md

README.md

@@ -71,17 +71,31 @@ On the held-out test set (20% of data):
 import joblib
 import xgboost as xgb
 from sentence_transformers import SentenceTransformer
+from huggingface_hub import hf_hub_download
 
-# Load model + embedder
-model = joblib.load("Ubuntu_xgb_model.pkl")
+# -----------------------------
+# 1. Download model from Hugging Face Hub
+# -----------------------------
+REPO_ID = "mjpsm/Ubuntu_xgb_model"  # change if you used a different repo name
+FILENAME = "Ubuntu_xgb_model.pkl"
+
+model_path = hf_hub_download(repo_id=REPO_ID, filename=FILENAME)
+
+# -----------------------------
+# 2. Load model + embedder
+# -----------------------------
+model = joblib.load(model_path)
 embedder = SentenceTransformer("all-mpnet-base-v2")
 
-# Example prediction
+# -----------------------------
+# 3. Example prediction
+# -----------------------------
 text = "During our class project, I made sure everyone’s ideas were included."
 embedding = embedder.encode([text])
 score = model.predict(embedding)[0]
 
-print("Predicted Ubuntu Score:", score)
+print("Predicted Ubuntu Score:", round(float(score), 3))
+
 ```
 
 ## 🌍 Applications