Update main.py

main.py

@@ -35,10 +35,23 @@ class StudentFeatures(BaseModel):
 # ==============================
 # 3. Muat model (kompatibel dengan skops >= 0.10)
 # ==============================
-# Pastikan file model_risiko_akademik.skops berada di direktori yang sama dengan file ini.
 MODEL_PATH = os.path.join(os.path.dirname(__file__), "model_risiko_akademik.skops")
+
 try:
-    model = sio.load(MODEL_PATH, trusted=True)
+    trusted_types = [
+        "numpy.ndarray",
+        "numpy.core.multiarray.scalar",
+        "sklearn.tree._classes.DecisionTreeClassifier",
+        "_codecs.encode",
+        "joblib.numpy_pickle.NumpyArrayWrapper",
+        "numpy.core.multiarray._reconstruct",
+        "numpy.dtype",
+        "sklearn.tree._tree.Tree"
+    ]
+    
+    # Muat model dengan daftar tipe yang sudah didefinisikan secara eksplisit
+    model = sio.load(MODEL_PATH, trusted=trusted_types)
+
 except Exception as e:
     raise RuntimeError(f"Gagal memuat model dari {MODEL_PATH}: {e}")