Rahul23232 commited on
Commit
760a5f3
·
verified ·
1 Parent(s): c998ffa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +27 -0
app.py CHANGED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import pickle
3
+
4
+ # Load model
5
+ model = pickle.load(open("model.pkl", "rb"))
6
+
7
+ flowers = ["Setosa", "Versicolor", "Virginica"]
8
+
9
+ def predict(sepal_length, sepal_width, petal_length, petal_width):
10
+ data = [[sepal_length, sepal_width, petal_length, petal_width]]
11
+ pred = model.predict(data)[0]
12
+ return flowers[pred]
13
+
14
+ # UI
15
+ interface = gr.Interface(
16
+ fn=predict,
17
+ inputs=[
18
+ gr.Number(label="Sepal Length"),
19
+ gr.Number(label="Sepal Width"),
20
+ gr.Number(label="Petal Length"),
21
+ gr.Number(label="Petal Width")
22
+ ],
23
+ outputs="text",
24
+ title="🌸 Flower Prediction App"
25
+ )
26
+
27
+ interface.launch()