|
import pickle |
|
import gradio as gr |
|
def make_predicition(sepal_length,sepal_width,petal_length,petal_width): |
|
with open("model_iris.pkl","rb") as f: |
|
clf=pickle.load(f) |
|
preds=clf.predict([[sepal_length,sepal_width,petal_length,petal_width]]) |
|
if preds==0: |
|
return "IRIS-SETOSA" |
|
elif preds==1: |
|
return "IRIS-VERSICOLOR" |
|
else: |
|
return "IRIS-VIRGINICA" |
|
|
|
sep_len=gr.Number(label="SEPAL LENGTH") |
|
sep_wid=gr.Number(label="SEPAL WIDTH") |
|
pet_len=gr.Number(label="PETAL LENGTH") |
|
pet_wid=gr.Number(label="PETAL WIDTH") |
|
|
|
output=gr.Textbox() |
|
|
|
app=gr.Interface(fn=make_predicition,inputs=[sep_len,sep_wid,pet_len,pet_wid],outputs=output) |
|
app.launch() |