Hugging Face's logo

Spaces:
NicolasvonRotz
/
Lego-Bricks-AI
Runtime error

App Files Community
NicolasvonRotz commited on
Commit
ea4dfb9
·
1 Parent(s): eab1c21

added new test data

Browse files
Files changed (7) hide show
  1. 0002.png +0 -0
  2. 18654 beam 1M 259R.png +0 -0
  3. 201706161606-0002.png +0 -0
  4. 201706161906-0001.png +0 -0
  5. 2357 brick corner 1x2x2 006L.png +0 -0
  6. 3046 roof corner inside tile 2x2 272L.png +0 -0
  7. app.py +1 -1
0002.png ADDED
18654 beam 1M 259R.png DELETED
Binary file (11.1 kB)
 
201706161606-0002.png ADDED
201706161906-0001.png ADDED
2357 brick corner 1x2x2 006L.png DELETED
Binary file (31.8 kB)
 
3046 roof corner inside tile 2x2 272L.png DELETED
Binary file (43.4 kB)
 
app.py CHANGED
@@ -5,7 +5,7 @@ import gradio as gr
5
  def classify_image(img):
6
  from fastai.vision.all import load_learner
7
  learn = load_learner('lego-bricks-model.pkl')
8
- categories = ('3046 roof corner inside tile 2x2', '2357 brick corner 1x2x2x', '18654 beam 1M')
9
  pred,idx,probs = learn.predict(img)
10
  return dict(zip(categories, map(float,probs)))
11
 
 
5
  def classify_image(img):
6
  from fastai.vision.all import load_learner
7
  learn = load_learner('lego-bricks-model.pkl')
8
+ categories = ('3022 Plate 2X2', '3673 Peg 2M', '2794 Plate 1X2with 1 Knob')
9
  pred,idx,probs = learn.predict(img)
10
  return dict(zip(categories, map(float,probs)))
11