VivekSivaramakrishnan commited on
Commit
bda383b
·
1 Parent(s): cd2d6a3
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ def classify_image(img):
8
  pred, idx, probs = learn.predict(img)
9
  return dict(zip(categories, map(float,probs)))
10
 
11
- image = gr.Image(shape=(256, 256))
12
  label = gr.Label()
13
  examples = ['doughnut.jpg', 'vada.jpg']
14
 
 
8
  pred, idx, probs = learn.predict(img)
9
  return dict(zip(categories, map(float,probs)))
10
 
11
+ image = gr.Image(height=256, width=256)
12
  label = gr.Label()
13
  examples = ['doughnut.jpg', 'vada.jpg']
14