ejazhabibdar commited on
Commit
6610283
1 Parent(s): a392e4a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -52,8 +52,8 @@ def classify_image(img):
52
  category = categories[idx]
53
  return dict(zip(categories,map(float,probs)))
54
 
55
- image = gr.inputs.Image(shape=(192, 192))
56
- label = gr.outputs.Label()
57
 
58
  examples = ['image_07751.jpg', 'image_04897.jpg',
59
  'image_02553.jpg', 'image_06930.jpg',
 
52
  category = categories[idx]
53
  return dict(zip(categories,map(float,probs)))
54
 
55
+ image=gr.Image(type="pil")
56
+ label=gr.Label(num_top_classes=102)
57
 
58
  examples = ['image_07751.jpg', 'image_04897.jpg',
59
  'image_02553.jpg', 'image_06930.jpg',