Spaces:
Build error
Build error
| from fastai.vision.all import * | |
| import gradio as gr | |
| learn = load_learner('archi_styles_model.pkl') | |
| styles = ('Baroque Architecture','Gothic Architecture', 'Greek and Roman Classical Architecture', 'Modern Architecture', 'Neoclassical Architecture', 'Neofuturist Architecture', 'Post-Modern Architecture', 'Victorian Architecture') | |
| examples = ['Baroque Architecture.jpg', 'Gothic Architecture.jpg', 'Modern Architecture.jpg'] | |
| # Use dict() and zip() to build a dict, not formaliar with data object in python. | |
| def image_classifier(img): | |
| pred_styles,pred_idx,probs = learn.predict(img) | |
| return dict(zip(styles, map(float, probs))) | |
| demo = gr.Interface(fn=image_classifier, inputs="image", outputs="label", examples = examples) | |
| demo.launch() |