Update app.py
Browse files
app.py
CHANGED
@@ -13,8 +13,8 @@ def predict_image(img):
|
|
13 |
|
14 |
model = VGG16()
|
15 |
model.summary()
|
16 |
-
image = gr.
|
17 |
-
label = gr.
|
18 |
|
19 |
gr.Interface(fn=predict_image,
|
20 |
title="VGG16 Classification",
|
|
|
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,
|
20 |
title="VGG16 Classification",
|