| 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() | |