Tabas34/electronics-recogniser

1from fastai.vision.all import load_learner2import gradio as gr3 4elec_labels=(5    'Air Conditioner appliances',6    'Air Fryer appliances',7    'Blender appliances',8    'Clothes Dryer appliances',9    'Coffee Maker appliances',10    'Dish Washer appliances',11    'Electric Guitar appliances',12    'Electric toothbrush appliances',13    'Fan appliances',14    'Griller appliances',15    'Hair dryer appliances',16    'Heater appliances',17    'Induction cooktop appliances ',18    'Iron appliances',19    'Kettle appliances',20    'Microwave appliances',21    'Mixer appliances',22    'Refrigerator appliances',23    'Rice Cooker appliances',24    'Speaker appliances',25    'Toaster appliances',26    'Vacuum Cleaner appliances'27)28 29model=load_learner(f'models/elec-recognizer-v4.pkl')30 31 32def recognize_image(image):33  # image = input_image.resize((192, 192))34  pred,idx,probs=model.predict(image)35  return dict(zip(elec_labels,map(float,probs)))36 37 38 39image = gr.Image()40label = gr.Label()41examples=[42    'test_images/unknown_00.jpg',43    'test_images/unknown_01.jpg.webp',44    'test_images/unknown_02.webp',45    'test_images/unknown_03.jpeg',46    'test_images/unknown_04.jpg'47]48 49iface = gr.Interface(fn=recognize_image, inputs=image, outputs=label,examples=examples)50iface.launch(inline=False)