Sync App files

README.md

@@ -1,11 +1,3 @@
----
-license: apache-2.0
-title: drug-discovery
-sdk: gradio
-emoji: 📊
-colorFrom: green
-colorTo: blue
----
 title: Drug Classification
 emoji: 📉
 colorFrom: purple

app.py

@@ -1,8 +1,5 @@
 import gradio as gr
-import joblib
-
-
-
+import skops.io as sio
 
 # Explicitly specify trusted types
 trusted_types = [
@@ -14,11 +11,11 @@ trusted_types = [
     "sklearn.impute.SimpleImputer",
     "sklearn.tree.DecisionTreeClassifier",
     "sklearn.ensemble.RandomForestClassifier",
-    "numpy.dtype",
+    "numpy.dtype",  # Add this line to trust numpy.dtype
 ]
 
-pipe = joblib.load("../Model/model.pkl")
-
+# Load the model using skops (adjust path to your .skops file)
+pipe = sio.load("../Model/drug_pipeline.skops", trusted=trusted_types)  # Use .skops file instead of .pkl
 
 def predict_drug(age, sex, blood_pressure, cholesterol, na_to_k_ratio):
     """Predict drugs based on patient features.