#!/usr/bin/env python # coding: utf-8 # In[1]: #|default_exp sala # In[2]: #|export from fastai.vision.all import * import gradio as gr def is_cat(x): return x[0].isupper() # In[3]: im= PILImage.create('dog.jpg') im.thumbnail((192,192)) im # In[4]: #|export import pathlib temp = pathlib.PosixPath pathlib.PosixPath = pathlib.WindowsPath # In[5]: #|export learn=load_learner('model.pkl') # In[6]: learn.predict(im) # In[7]: #|export categories=('dog','cat') def classify_image(img): pred,idx,probs=learn.predict(img) return dict(zip(categories, map(float,probs))) # In[8]: classify_image(im) # In[9]: #|export image=gr.inputs.Image(shape=(192,192)) label = gr.outputs.Label() examples=['dog.jpg','cat.jpg','dunno.jpg'] intf=gr.Interface(fn=classify_image,inputs=image,outputs=label,examples=examples) intf.launch(inline=False) # In[10]: