iris / app.py
Rahul23232's picture
Update app.py
d630bf1 verified
Raw
History Blame Contribute Delete
552 Bytes
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()