Spaces:

NicolasvonRotz
/

Lego-Bricks-AI

Runtime error

NicolasvonRotz commited on May 1, 2023

Commit

5bb2a8a

1 Parent(s): fa6f28c

new model import

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ import gradio as gr
 def classify_image(img):
     from fastai.vision.all import load_learner
-    learn = load_learner('lego-bricks-model.pkl')
     categories = (
         'Brick_2357', 'Brick_3002', 'Brick_3003', 'Brick_3004', 'Brick_3010', 'Brick_30145', 'Brick_3065', 'Brick_6223', 'Brick_702'
     )

 def classify_image(img):
     from fastai.vision.all import load_learner
+    learn = load_learner('lego-bricks-detecter-model.pkl')
     categories = (
         'Brick_2357', 'Brick_3002', 'Brick_3003', 'Brick_3004', 'Brick_3010', 'Brick_30145', 'Brick_3065', 'Brick_6223', 'Brick_702'
     )