Rahul23232 commited on
Commit
eeadd9b
·
verified ·
1 Parent(s): 113245f

Create app.py

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