KrishnaRDL/stringinstrumentsv2

1import gradio as gr2from fastai.vision.all import *3import timm4import skimage5 6learn = load_learner('stringinstrumentsv2.pkl')7 8labels = learn.dls.vocab9def predict(img):10    img = PILImage.create(img)11    pred,pred_idx,probs = learn.predict(img)12    return {labels[i]: float(probs[i]) for i in range(len(labels))}13 14#title = "String Instrument Classifier"15#description = "A string instrument classifier trained on a DDG image search dataset with fastai."16#examples = ['doublebass.jpg']17#interpretation='default'18#enable_queue=True19 20#gr.Interface(fn=predict,inputs=gr.Image(shape=(512, 512)),outputs=gr.Label(num_top_classes=4),title=title,description=description,examples=examples,interpretation=interpretation,enable_queue=enable_queue).launch()21 22#gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(512, 512)), outputs=gr.outputs.Label(num_top_classes=4)).launch(share=False)23 24gr.Interface(fn=predict, inputs=gr.Image(shape=(512, 512)), outputs=gr.Label(num_top_classes=4)).launch(share=False)