color model

app.py

@@ -4,9 +4,9 @@ import gradio as gr
 
 def classify_image(img):
     from fastai.vision.all import load_learner
-    learn = load_learner('lego-bricks-detecter-model.pkl')
+    learn = load_learner('lego-bricks-detecter-model-color.pkl')
     categories = (
-        'Brick_2357', 'Brick_3002', 'Brick_3003', 'Brick_3004', 'Brick_3010', 'Brick_30145', 'Brick_3065', 'Brick_6223', 'Brick_702'
+        'Color_1', 'Color_10', 'Color_100', 'Color_11', 'Color_110', 'Color_112', 'Color_114', 'Color_115', 'Color_117', 'Color_118', 'Color_12', 'Color_120', 'Color_125', 'Color_128', 'Color_129', 'Color_13', 'Color_132', 'Color_133', 'Color_134', 'Color_135', 'Color_137', 'Color_14', 'Color_142', 'Color_148', 'Color_15', 'Color_150', 'Color_151', 'Color_16', 'Color_17', 'Color_178', 'Color_179', 'Color_18', 'Color_183', 'Color_184', 'Color_186', 'Color_189', 'Color_19', 'Color_191', 'Color_2', 'Color_20', 'Color_21', 'Color_212', 'Color_216', 'Color_218', 'Color_219', 'Color_22', 'Color_226', 'Color_23', 'Color_231', 'Color_232', 'Color_234', 'Color_25', 'Color_256', 'Color_26', 'Color_27', 'Color_272', 'Color_28', 'Color_285', 'Color_288', 'Color_29', 'Color_293', 'Color_294', 'Color_295', 'Color_297', 'Color_3', 'Color_30', 'Color_302', 'Color_308', 'Color_31', 'Color_313', 'Color_32', 'Color_320', 'Color_321', 'Color_322', 'Color_323', 'Color_326', 'Color_329', 'Color_33', 'Color_330', 'Color_334', 'Color_335', 'Color_339', 'Color_34', 'Color_341', 'Color_35', 'Color_351', 'Color_353', 'Color_36', 'Color_360', 'Color_362', 'Color_364', 'Color_366', 'Color_368', 'Color_37', 'Color_370', 'Color_373', 'Color_378', 'Color_379', 'Color_38', 'Color_383', 'Color_39', 'Color_4', 'Color_40', 'Color_41', 'Color_42', 'Color_43', 'Color_44', 'Color_45', 'Color_450', 'Color_46', 'Color_462', 'Color_47', 'Color_484', 'Color_5', 'Color_503', 'Color_508', 'Color_509', 'Color_510', 'Color_52', 'Color_54', 'Color_57', 'Color_6', 'Color_60', 'Color_61', 'Color_62', 'Color_63', 'Color_64', 'Color_68', 'Color_69', 'Color_7', 'Color_70', 'Color_71', 'Color_72', 'Color_73', 'Color_74', 'Color_75', 'Color_76', 'Color_77', 'Color_78', 'Color_79', 'Color_8', 'Color_80', 'Color_81', 'Color_82', 'Color_83', 'Color_84', 'Color_85', 'Color_86', 'Color_89', 'Color_9', 'Color_92'
     )
 
     pred,idx,probs = learn.predict(img)
@@ -18,4 +18,3 @@ examples = []
 
 intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)
 intf.launch(inline=False)
- 

lego-bricks-detecter-model-color.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6e1cadb8b78bb32b4fd729d327bd0bd30168a751665d63757d6e845185850eae
+size 201452908