Classifier / app.py
Daraphan's picture
Create app.py
3ad0825 verified
raw
history blame contribute delete
796 Bytes
from fastai.vision.all import load_learner
import gradio as gr
labels=(
'African people picture',
'British people picture',
'East Asain people picture',
'Eskimo people picture',
'European people picture',
'Indigenous American people picture',
'Indigenous Australians people picture',
'Indigenous people picture',
'Negrito people picture',
'Semitic people picture'
)
model=load_learner(f'humanRace-v11.pkl')
def recognize_image(image):
pred, idx, probs=model.predict(image)
print(pred)
return dict(zip(labels, map(float, probs)))
image = gr.inputs.Image(image_mode="RGB")
label=gr.outputs.Label()
example=['th(11).jpeg']
iface = gr.Interface(fn=recognize_image, inputs=image, outputs=label,examples=example )
iface.launch(inline=False, share=True)