Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| import pickle | |
| # Load model | |
| model = pickle.load(open("model.pkl", "rb")) | |
| flowers = ["Setosa", "Versicolor", "Virginica"] | |
| def predict(sepal_length, sepal_width, petal_length, petal_width): | |
| data = [[sepal_length, sepal_width, petal_length, petal_width]] | |
| pred = model.predict(data)[0] | |
| return flowers[pred] | |
| # UI | |
| interface = gr.Interface( | |
| fn=predict, | |
| inputs=[ | |
| gr.Number(label="Sepal Length"), | |
| gr.Number(label="Sepal Width"), | |
| gr.Number(label="Petal Length"), | |
| gr.Number(label="Petal Width") | |
| ], | |
| outputs="text", | |
| title="🌸 Flower Prediction App" | |
| ) | |
| interface.launch() |