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()