Spaces:
Sleeping
Sleeping
File size: 655 Bytes
760a5f3 2fb91f5 760a5f3 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 | 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() |