KaranNag commited on
Commit
d8840d1
1 Parent(s): 59c7ba1

updating classes

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -66,7 +66,7 @@ example_list = [["examples/" + example] for example in os.listdir("examples")]
66
  # Create the Gradio demo
67
  demo = gr.Interface(fn=predict, # mapping function from input to output
68
  inputs=gr.Image(type="pil"), # what are the inputs?
69
- outputs=[gr.Label(num_top_classes=3, label="Predictions"), # what are the outputs?
70
  gr.Number(label="Prediction time (s)")], # our fn has two outputs, therefore we have two outputs
71
  # Create examples list from "examples/" directory
72
  examples=example_list,
 
66
  # Create the Gradio demo
67
  demo = gr.Interface(fn=predict, # mapping function from input to output
68
  inputs=gr.Image(type="pil"), # what are the inputs?
69
+ outputs=[gr.Label(num_top_classes=10, label="Predictions"), # what are the outputs?
70
  gr.Number(label="Prediction time (s)")], # our fn has two outputs, therefore we have two outputs
71
  # Create examples list from "examples/" directory
72
  examples=example_list,