right app

app.py

@@ -4,9 +4,8 @@ import gradio as gr
 
 def classify_image(img):
     from fastai.vision.all import load_learner
-    learn = load_learner('bricks-model.pkl')
-    categories = ( 'Brick_11103', 'Brick_112', 'Brick_11262', 'Brick_11833', 'Brick_149', 'Brick_15265', 'Brick_15400', 'Brick_18946', 'Brick_18973', 'Brick_22472', 'Brick_2357', 'Brick_242021', 'Brick_2432', 'Brick_24581', 'Brick_2467', 'Brick_25126', 'Brick_2536', 'Brick_25561', 'Brick_26021', 'Brick_2610', 'Brick_2696', 'Brick_2737', 'Brick_2804', 'Brick_2807', 'Brick_29124', 'Brick_3002', 'Brick_3003', 'Brick_3004', 'Brick_3010', 'Brick_30145', 'Brick_30152', 'Brick_30250', 'Brick_30277', 'Brick_30284', 'Brick_30303', 'Brick_3033', 'Brick_30409', 'Brick_30505', 'Brick_3065', 'Brick_3070', 'Brick_32065', 'Brick_32138', 'Brick_32324', 'Brick_32373', 'Brick_32555', 'Brick_33048', 'Brick_3460', 'Brick_35565', 'Brick_3828', 'Brick_3844', 'Brick_389', 'Brick_3898', 'Brick_40341', 'Brick_41667', 'Brick_4211622', 'Brick_4247', 'Brick_42603', 'Brick_4336', 'Brick_43802', 'Brick_44137', 'Brick_4582471', 'Brick_4599', 'Brick_46225', 'Brick_47994', 'Brick_50304', 'Brick_55704', 'Brick_56220', 'Brick_58120', 'Brick_60621', 'Brick_60657', 'Brick_6075', 'Brick_61506', 'Brick_6190', 'Brick_61905', 'Brick_6214', 'Brick_6223', 'Brick_62622', 'Brick_630', 'Brick_63869', 'Brick_64681', 'Brick_6517', 'Brick_67704', 'Brick_6942', 'Brick_702', 'Brick_72', 'Brick_727', 'Brick_74040', 'Brick_744', 'Brick_74880', 'Brick_76270', 'Brick_76607', 'Brick_80298', 'Brick_88326', 'Brick_89650', 'Brick_90616', 'Brick_9244', 'Brick_9352', 'Brick_94638', 'Brick_97782', 'Brick_98286', 'Brick_99013' )
-
+    learn = load_learner('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')
     pred,idx,probs = learn.predict(img)
     return dict(zip(categories, map(float,probs)))