devudilip commited on
Commit
71f8810
·
verified ·
1 Parent(s): 5ee37c0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -13,12 +13,12 @@ def classify_image(img):
13
  pred, idx, probs = learn.predict(img)
14
  return dict(zip(categories, map(float, probs)))
15
 
16
- image = gr.Image()
17
  label = gr.Label()
18
 
19
  examples = ['siamese.png']
20
 
21
- intf = gr.Interface(fn=classify_image, inputs=image, outputs=label)
22
  intf.launch(inline=False)
23
 
24
  # def greet(name, intensity):
 
13
  pred, idx, probs = learn.predict(img)
14
  return dict(zip(categories, map(float, probs)))
15
 
16
+ image = gr.Image(image_mode="RGB", type="pil")
17
  label = gr.Label()
18
 
19
  examples = ['siamese.png']
20
 
21
+ intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)
22
  intf.launch(inline=False)
23
 
24
  # def greet(name, intensity):