ipd commited on
Commit
2ac89d4
1 Parent(s): e2893d6
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -48,14 +48,14 @@ def predict(Image):
48
 
49
 
50
 
51
- #image = gr.Image(shape=(224, 224), image_mode="RGB")
52
  #label = gr.Label(label="Grade")
53
 
54
  demo = gr.Interface(
55
  fn=predict,
56
- inputs="image",
57
- outputs="image",#label,
58
- examples=["examples/0.png", "examples/1.png", "examples/2.png", "examples/3.png", "examples/4.png"]
59
  )
60
 
61
  if __name__ == "__main__":
 
48
 
49
 
50
 
51
+ image = gr.Image(type="pil")#shape=(224, 224), image_mode="RGB")
52
  #label = gr.Label(label="Grade")
53
 
54
  demo = gr.Interface(
55
  fn=predict,
56
+ inputs=image,
57
+ outputs=image,#label,
58
+ #examples=["examples/0.png", "examples/1.png", "examples/2.png", "examples/3.png", "examples/4.png"]
59
  )
60
 
61
  if __name__ == "__main__":