| import gradio as gr |
| import skops.io as sio |
|
|
| pipe = sio.load("./Model/drug_pipeline.skops", trusted=True) |
|
|
|
|
| def predict_drug(): |
| predicted_drug = pipe.score() |
|
|
| |
| return predicted_drug |
|
|
|
|
| outputs = [gr.Label(num_top_classes=5)] |
|
|
|
|
|
|
| title = "Drug Classification" |
| description = "Enter the details to correctly identify Drug type?" |
| article = "This app is a part of the Beginner's Guide to CI/CD for Machine Learning. It teaches how to automate training, evaluation, and deployment of models to Hugging Face using GitHub Actions." |
|
|
|
|
| gr.Interface( |
| fn=predict_drug, |
| outputs=outputs, |
| title=title, |
| description=description, |
| article=article, |
| theme=gr.themes.Soft(), |
| ).launch() |
|
|
