marufc36
add app.py
cc966a6
raw
history blame
800 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'models/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)