Spaces:
Runtime error
Runtime error
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=['th(11).jpeg'] | |
iface = gr.Interface(fn=recognize_image, inputs=image, outputs=label,examples=example ) | |
iface.launch(inline=False, share=True) | |