File size: 954 Bytes
21673ed 3b5ebda 21673ed 3b5ebda 21673ed |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 |
from fastai.vision.all import load_learner
import gradio as gr
musical_instruments = (
'Bansuri',
'Harmonium',
'Pakhawaz',
'Santoor',
'Sarangi',
'Sarod',
'Shehnai',
'Sitar',
'Tabla',
'Tanpura'
)
model = load_learner(f"models/musical-instrument-recognizer-v6.pkl")
def recognize_image(image):
pred, idx, probs = model.predict(image)
return dict(zip(musical_instruments, map(float, probs)))
image = gr.inputs.Image(shape=(192,192))
label = gr.outputs.Label()
examples = [
'test_data/bansuri.jfif',
'test_data/harmonium.jfif',
'test_data/pakhawaj.jfif',
'test_data/santoor.jfif',
'test_data/sarengi.jfif',
'test_data/sarod.jfif',
'test_data/shehnai.jfif',
'test_data/sitar.jfif',
'test_data/tabla.jfif',
'test_data/tanpura.jfif'
]
iface = gr.Interface(fn=recognize_image, inputs = image, outputs= label, examples = examples)
iface.launch(inline =False, share = True) |