anatexis commited on
Commit
277d88e
1 Parent(s): 8906f61

changed predict function

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,5 +20,5 @@ examples = ['ninjago.jpeg', 'lego.jpeg']
20
  interpretation = 'default'
21
  enable_queue = True
22
 
23
- gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(512, 512)), outputs=gr.outputs.Label(num_top_classes=2), title=title,
24
  description=description, examples=examples, interpretation=interpretation, enable_queue=enable_queue).launch()
 
20
  interpretation = 'default'
21
  enable_queue = True
22
 
23
+ gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(192, 192)), outputs=gr.outputs.Label(), title=title,
24
  description=description, examples=examples, interpretation=interpretation, enable_queue=enable_queue).launch()