g0blas commited on
Commit
849e970
1 Parent(s): e0ad037

updated app

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ model = load_learner(model_path)
30
 
31
  def recognize_image(image):
32
  pred, idx, probs = model.predict(image)
33
- return dict(zip(cap_labels, map(float, probs)))
34
 
35
  image = gr.inputs.Image(shape=(192, 192))
36
  label = gr.outputs.Label()
 
30
 
31
  def recognize_image(image):
32
  pred, idx, probs = model.predict(image)
33
+ return dict(zip(chicken_breeds, map(float, probs)))
34
 
35
  image = gr.inputs.Image(shape=(192, 192))
36
  label = gr.outputs.Label()