xuyaxiong commited on
Commit
d331990
1 Parent(s): 0b06928
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -1,5 +1,6 @@
1
  from fastai.vision.all import *
2
  import gradio as gr
 
3
  import pathlib
4
 
5
  plt = platform.system()
@@ -13,8 +14,8 @@ def classify_image(img):
13
  pred,idx,probs = learn.predict(img)
14
  return dict(zip(categories, map(float, probs)))
15
 
16
- image = gr.inputs.Image(shape=(28,28))
17
- label = gr.outputs.Label()
18
  examples = ['img_0.png', 'img_1.jpg']
19
 
20
  intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)
 
1
  from fastai.vision.all import *
2
  import gradio as gr
3
+ from gradio.components import Image, Label
4
  import pathlib
5
 
6
  plt = platform.system()
 
14
  pred,idx,probs = learn.predict(img)
15
  return dict(zip(categories, map(float, probs)))
16
 
17
+ image = Image(shape=(28,28))
18
+ label = Label()
19
  examples = ['img_0.png', 'img_1.jpg']
20
 
21
  intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)