panik commited on
Commit
8891d61
·
verified ·
1 Parent(s): 47eec39

remove image shape parameter

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ def predict_image(img):
13
 
14
  model = VGG16()
15
  model.summary()
16
- image = gr.Image(shape=(224,224))
17
  label = gr.Label(num_top_classes=5)
18
 
19
  gr.Interface(fn=predict_image,
 
13
 
14
  model = VGG16()
15
  model.summary()
16
+ image = gr.Image() # shape=(224,224))
17
  label = gr.Label(num_top_classes=5)
18
 
19
  gr.Interface(fn=predict_image,