fix the untrsuted issues

kingabzpro · Dec 2, 2024 · e52027a · e52027a · github-actions · Dec 2, 2024
1 parent 2c7feff
commit e52027a
Showing 1 changed file with 18 additions and 1 deletion.
diff --git a/App/drug_app.py b/App/drug_app.py
@@ -1,7 +1,24 @@
 import gradio as gr
 import skops.io as sio
+import warnings
+from sklearn.exceptions import InconsistentVersionWarning
 
-pipe = sio.load("./Model/drug_pipeline.skops", trusted=False)
+# Suppress the version warnings
+warnings.filterwarnings("ignore", category=InconsistentVersionWarning)
+
+# Explicitly specify trusted types
+trusted_types = [
+    "sklearn.pipeline.Pipeline",
+    "sklearn.preprocessing.OneHotEncoder",
+    "sklearn.preprocessing.StandardScaler",
+    "sklearn.compose.ColumnTransformer",
+    "sklearn.preprocessing.OrdinalEncoder",
+    "sklearn.impute.SimpleImputer",
+    "sklearn.tree.DecisionTreeClassifier",
+    "sklearn.ensemble.RandomForestClassifier",
+    "numpy.dtype",
+]
+pipe = sio.load("./Model/drug_pipeline.skops", trusted=trusted_types)
 
 
 def predict_drug(age, sex, blood_pressure, cholesterol, na_to_k_ratio):