aaalto commited on
Commit
f42e31a
1 Parent(s): 2c6d1f1

Fix categories

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import gradio as gr
4
  learn = load_learner('model.pkl')
5
 
6
  #! export
7
- categories = ('Ahrend Result Chair', 'Borge Mogenson J39 Chair', 'Breuer Cesca Chair', 'PK22 Chair', 'Series 7 Chair', 'Vitra Standard Chair')
8
  def classify_image(img):
9
  pred, idx, probs = learn.predict(img)
10
  return dict(zip(categories, map(float, probs)))
 
4
  learn = load_learner('model.pkl')
5
 
6
  #! export
7
+ categories = ('Ahrend Result Chair', 'Borge Mogenson J39 Chair', 'Breuer Cesca Chair', 'Jean prouve Vitra Standard Chair', 'PK22 Chair Poul Kjærholm', 'Series 7 Chair Arne Jacobsen', )
8
  def classify_image(img):
9
  pred, idx, probs = learn.predict(img)
10
  return dict(zip(categories, map(float, probs)))