File size: 1,263 Bytes
a2055dc c98afd9 04208ab |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
__all__ = ['learn', 'classify_image', 'categories', 'image', 'label', 'examples', 'intf']
from fastai.vision.all import *
import gradio as gr
learn = load_learner('architecturemodel.pkl')
categories = ('Achaemenid architecture',
'American Foursquare architecture',
'American craftsman style',
'Ancient Egyptian architecture',
'Art Deco architecture',
'Art Nouveau architecture',
'Baroque architecture',
'Bauhaus architecture',
'Beaux-Arts architecture',
'Byzantine architecture',
'Chicago school architecture',
'Colonial architecture',
'Deconstructivism',
'Edwardian architecture',
'Georgian architecture',
'Gothic architecture',
'Greek Revival architecture',
'International style',
'Novelty architecture',
'Palladian architecture',
'Postmodern architecture',
'Queen Anne architecture',
'Romanesque architecture',
'Russian Revival architecture',
'Tudor Revival architecture')
def classify_image(img):
pred,idx,probs = learn.predict(img)
return dict(zip(categories, map(float,probs)))
image = gr.inputs.Image(shape=(192, 192))
label = gr.outputs.Label()
examples = ['bigben.jpeg','pyramid.jpeg','robiehouse.jpeg']
intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)
intf.launch(inline=False) |