sharathprasaath's picture
Update app.py
18ac803
raw
history blame contribute delete
573 Bytes
from fastai.vision.all import *
import gradio as gr
import pickle
# import pathlib
# temp = pathlib.PosixPath
# pathlib.PosixPath = pathlib.WindowsPath
learn=load_learner('model (1).pkl')
categories=('man','woman')
def classify(img):
pred,idx,prob=learn.predict(img)
return dict(zip(categories,map(float,prob)))
image=gr.inputs.Image(shape=(192,192))
label=gr.outputs.Label()
examples = ["man.jpg", "woman.jpg"]
intf = gr.Interface(fn=classify, inputs=image,
outputs=label, examples=examples, title="Man or Woman !!!")
intf.launch(inline=False)