model init

app.py

@@ -4,10 +4,8 @@ import gradio as gr
 
 def classify_image(img):
     from fastai.vision.all import load_learner
-    learn = load_learner('lego-bricks-detecter-model-color.pkl')
-    categories = (
-        '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'
-    )
+    learn = load_learner('bricks-model.pkl')
+    categories = ( 'Brick_242021', 'Brick_2467', 'Brick_2610', 'Brick_2804', 'Brick_30152', 'Brick_30277', 'Brick_32065', 'Brick_58120', 'Brick_60657', 'Brick_6942', 'Brick_89650' )
 
     pred,idx,probs = learn.predict(img)
     return dict(zip(categories, map(float,probs)))

bricks-model.pkl

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