nasrin2023ripa/nut-recognizer

1from fastai.vision.all import *2import gradio as gr3from pathlib import Path4 5#import pathlib6#temp = pathlib.PosixPath7#pathlib.PosixPath = pathlib.WindowsPath8 9nut_labels = [10    'raw Almonds', 11    'raw Brazil nut', 12    'raw Cashew nut', 13    'raw Chestnut', 14    'raw Ginkgo nut', 15    'raw Hazelnuts', 16    'raw Hickory nut', 17    'raw Maccademia nut', 18    'raw Peanut', 19    'raw Pecans', 20    'raw Pili nut', 21    'raw Pine nut', 22    'raw Pistachios nut', 23    'raw Walnuts'24    ]25 26model = load_learner('nut-recognizer-v14 .pkl')27 28 29def recognize_image(image):30    pred, idx, probs = model.predict(image)31    return dict(zip(nut_labels, map(float, probs)))32 33image = gr.inputs.Image(shape=(192,192))34label = gr.outputs.Label(num_top_classes=5)35examples = [36    'pistachios.jpeg',37    'cashew.jpg',38    'macadamia.jpg',39    'walnuts.jpg',40    'chestnuts.jpg',41    'pili nuts.jpg',42    'ginkgo nuts.jpg'43    ]44 45iface = gr.Interface(fn=recognize_image, inputs=image, outputs=label, examples=examples)46iface.launch(inline=False)47