Sync App files

drug_app.py

@@ -1,14 +1,31 @@
 import gradio as gr
+import skops.io as sio
 
+pipe = sio.load("./Model/drug_pipeline.skops", trusted=True)
 
-def greet(name, intensity):
-    return "Hello " * intensity + name + "!"
 
+def classifier(Age, Sex, BP, Cholesterol, Na_to_K):
+    pred_glass = pipe.predict([[Age, Sex, BP, Cholesterol, Na_to_K]])[0]
+    label = f"Predicted Glass label: **{pred_glass}**"
+    return label
 
-demo = gr.Interface(
-    fn=greet,
-    inputs=["text", "slider"],
-    outputs=["text"],
-)
 
-demo.launch()
+inputs = [
+    gr.Slider(15, 74, step=1, label="Age"),
+    gr.Radio(["M", "F"], label="Sex"),
+    gr.Radio(["HIGH", "LOW", "NORMAL"], label="Blood Pressure"),
+    gr.Radio(["HIGH", "NORMAL"], label="Cholesterol"),
+    gr.Slider(6.2, 38.2, step=0.1, label="Na_to_K"),
+]
+outputs = [gr.Label(num_top_classes=5)]
+
+title = "Drug Classification"
+description = "Enter the details to correctly identify Drug type?"
+
+gr.Interface(
+    fn=classifier,
+    inputs=inputs,
+    outputs=outputs,
+    title=title,
+    description=description,
+).launch()