Rahul23232's picture
Update app.py
2fb91f5 verified
raw
history blame contribute delete
655 Bytes
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()