marufc36
update app.py
593be4c
raw
history blame
793 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.Image(image_mode="RGB")
label=gr.Label()
example=['test_images/th(11).jpeg']
iface = gr.Interface(fn=recognize_image, inputs=image, outputs=label,examples=example )
iface.launch(inline=False, share=True)