djsnuggz commited on
Commit
dadcac2
·
1 Parent(s): a730f4e
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ def classify_image(img):
9
  pred,idx,probs = learn.predict(img)
10
  return dict(zip(categories, map(float, probs)))
11
 
12
- image = gr.Image(shape=(192,192))
13
  label = gr.Label()
14
  examples = ['trump.jpg', 'baldwin.jpg', 'closecall.jpg']
15
 
 
9
  pred,idx,probs = learn.predict(img)
10
  return dict(zip(categories, map(float, probs)))
11
 
12
+ image = gr.Image(height=192, width=192)
13
  label = gr.Label()
14
  examples = ['trump.jpg', 'baldwin.jpg', 'closecall.jpg']
15