zachfster commited on
Commit
2acd01f
·
1 Parent(s): 2320f10

modify app

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ def classify_image(img):
15
  # return {categories[i] : float(probs[i]) for i in range(len(categories))}
16
 
17
  image = gr.inputs.Image(shape=(192,192))
18
- label = gr.outputs.Label(categories=7)
19
 
20
  intf = gr.Interface(fn=classify_image, inputs=image, outputs=label)
21
  intf.launch(inline=False)
 
15
  # return {categories[i] : float(probs[i]) for i in range(len(categories))}
16
 
17
  image = gr.inputs.Image(shape=(192,192))
18
+ label = gr.outputs.Label(num_top_classes=7)
19
 
20
  intf = gr.Interface(fn=classify_image, inputs=image, outputs=label)
21
  intf.launch(inline=False)