LeGenDaProg/DeepLearningBase

1# %%2#|export3from fastai.vision.all import*4import gradio as gr5 6def is_cat(x):7    return x[0].isupper()8 9# %%10im = PILImage.create('dog.jpg')11im.thumbnail((192,192))12im13 14# %%15#|export16learn = load_learner('model.pkl')17 18# %%19learn.predict(im)20 21# %%22#|export23categories = ('Dog', 'Cat')24 25def classify_image(img):26    pred,idx,probs = learn.predict(img)27    return dict(zip(categories, map(float, probs)))28 29# %%30classify_image(im)31 32# %%33#|export34#  interface35image = gr.inputs.Image(shape=(192, 192))36label = gr.outputs.Label()37examples = ['dog.jpg', 'Cat.jpg', 'dunoo.jpg']38 39intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)40intf.launch(inline=False)41 42 43# %%44m = learn.model45 46# %%47ps = list(m.parameters())48 49# %%50ps[1]51 52# %%53 54 55 56