FarziBuilder/Local
0
1# AUTOGENERATED! DO NOT EDIT! File to edit: chickli.ipynb.2 3# %% auto 04__all__ = ['temp', 'learn', 'breeds', 'image', 'label', 'examples', 'intf', 'classify_image']5 6# %% chickli.ipynb 27#!pip install fastbook8import fastbook9from fastbook import *10#!pip install fastai11from fastai.vision.widgets import *12#!pip install gradio13import gradio as gr14 15import IPython16from IPython.display import display17from PIL import Image18 19def search_images(term, max_images=50):20 print(f"Searching for '{term}'")21 return search_images_ddg(term, max_images)22 23# %% chickli.ipynb 324import pathlib25#temp = pathlib.PosixPath26#pathlib.PosixPath = pathlib.WindowsPath27plt = platform.system()28if plt == 'Linux': pathlib.WindowsPath = pathlib.PosixPath29 30learn = load_learner('model.pkl')31 32breeds = ('Labrador Retrievers','German Shepherds','Golden Retrievers','French Bulldogs','Bulldogs','Beagles','Poodles','Rottweilers','Chihuahua')33 34def classify_image(img):35 pred,idx,probs = learn.predict(img)36 #return dict(zip(breeds, map(float,probs)))37 return "This is " + pred38 39image = gr.components.Image()40label = gr.components.Label()41 42intf = gr.Interface(fn=classify_image, inputs=image, outputs=label)43intf.launch(inline=False)44 