Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| import pickle | |
| model = pickle.load(open("model.pkl", "rb")) | |
| flowers = ["Setosa", "Versicolor", "Virginica"] | |
| def predict(sl, sw, pl, pw): | |
| data = [[sl, sw, pl, pw]] | |
| pred = model.predict(data)[0] | |
| return flowers[pred] | |
| 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() |