Dimitre commited on
Commit
9afb8c5
·
1 Parent(s): c1c1c66

Debug model output shape

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -32,6 +32,5 @@ def predict_fn(image):
32
 
33
  iface = gr.Interface(fn=predict_fn,
34
  inputs=gr.Image(shape=(224, 224)),
35
- outputs=gr.Label(num_top_classes=5),
36
- examples=["apples.jpeg", "banana.jpeg", "car.jpeg"])
37
  iface.launch()
 
32
 
33
  iface = gr.Interface(fn=predict_fn,
34
  inputs=gr.Image(shape=(224, 224)),
35
+ outputs=gr.Label(num_top_classes=5))
 
36
  iface.launch()