Spaces:
Sleeping
Sleeping
File size: 552 Bytes
eeadd9b d630bf1 eeadd9b d630bf1 eeadd9b | 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 | 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() |