Commit
•
8400d07
1
Parent(s):
5d8c383
fad
Browse files
app.py
CHANGED
|
@@ -50,7 +50,7 @@ def classify_image(image):
|
|
| 50 |
|
| 51 |
|
| 52 |
# Input component
|
| 53 |
-
inputs = gr.Image(shape=(image_size, image_size))
|
| 54 |
|
| 55 |
# Output component
|
| 56 |
outputs = gr.outputs.Label(num_top_classes=3)
|
|
|
|
| 50 |
|
| 51 |
|
| 52 |
# Input component
|
| 53 |
+
inputs = gr.Image(shape=(image_size, image_size,channels))
|
| 54 |
|
| 55 |
# Output component
|
| 56 |
outputs = gr.outputs.Label(num_top_classes=3)
|