Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -18,7 +18,7 @@ trusted_types = [
|
|
18 |
"sklearn.ensemble.RandomForestClassifier",
|
19 |
"numpy.dtype",
|
20 |
]
|
21 |
-
pipe = sio.load("/
|
22 |
|
23 |
|
24 |
def predict_drug(age, sex, blood_pressure, cholesterol, na_to_k_ratio):
|
@@ -57,11 +57,18 @@ examples = [
|
|
57 |
]
|
58 |
|
59 |
|
|
|
|
|
|
|
|
|
60 |
|
61 |
gr.Interface(
|
62 |
fn=predict_drug,
|
63 |
inputs=inputs,
|
64 |
outputs=outputs,
|
65 |
examples=examples,
|
|
|
|
|
|
|
66 |
theme=gr.themes.Soft(),
|
67 |
).launch()
|
|
|
18 |
"sklearn.ensemble.RandomForestClassifier",
|
19 |
"numpy.dtype",
|
20 |
]
|
21 |
+
pipe = sio.load("./Model/drug_pipeline.skops", trusted=trusted_types)
|
22 |
|
23 |
|
24 |
def predict_drug(age, sex, blood_pressure, cholesterol, na_to_k_ratio):
|
|
|
57 |
]
|
58 |
|
59 |
|
60 |
+
title = "Drug Classification"
|
61 |
+
description = "Enter the details to correctly identify Drug type?"
|
62 |
+
article = "This app is a part of the **[Beginner's Guide to CI/CD for Machine Learning](https://www.datacamp.com/tutorial/ci-cd-for-machine-learning)**. It teaches how to automate training, evaluation, and deployment of models to Hugging Face using GitHub Actions."
|
63 |
+
|
64 |
|
65 |
gr.Interface(
|
66 |
fn=predict_drug,
|
67 |
inputs=inputs,
|
68 |
outputs=outputs,
|
69 |
examples=examples,
|
70 |
+
title=title,
|
71 |
+
description=description,
|
72 |
+
article=article,
|
73 |
theme=gr.themes.Soft(),
|
74 |
).launch()
|