Abhi commited on
Commit
ffc7759
·
1 Parent(s): 3d4bfe4
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ def classify_img(img):
11
  pred, idx, probs = learner.predict(img)
12
  return dict(zip(categories,map(float,probs)))
13
 
14
- image = gr.inputs.Image(shape=512,512)
15
  label = gr.outputs.Label()
16
  examples = ['teddy_bear.jpg']
17
  intf = gr.Interface(fn=classify_img, inputs=image, outputs=label,examples=examples)
 
11
  pred, idx, probs = learner.predict(img)
12
  return dict(zip(categories,map(float,probs)))
13
 
14
+ image = gr.inputs.Image(shape=(512,512))
15
  label = gr.outputs.Label()
16
  examples = ['teddy_bear.jpg']
17
  intf = gr.Interface(fn=classify_img, inputs=image, outputs=label,examples=examples)