Update app.py
Browse files
app.py
CHANGED
|
@@ -20,6 +20,11 @@ class_names = [
|
|
| 20 |
"Truck",
|
| 21 |
]
|
| 22 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 23 |
IMG_SIZE = 72
|
| 24 |
|
| 25 |
def infer(input_image):
|
|
@@ -42,7 +47,6 @@ def infer(input_image):
|
|
| 42 |
|
| 43 |
input = gr.inputs.Image(shape=(IMG_SIZE, IMG_SIZE))
|
| 44 |
output = [gr.outputs.Label()]
|
| 45 |
-
examples = [[], []]
|
| 46 |
title = "Image Classification using "
|
| 47 |
description = "Upload an image or select from examples to classify it.<br>The allowed classes are - Airplane, Automobile, Bird, Cat, Deer, Dog, Frog, Horse, Ship, Truck.<br><p><b>Space author: Vivek Rai</b> <br><b> Keras example author: Sayak Paul </b></p>"
|
| 48 |
|
|
|
|
| 20 |
"Truck",
|
| 21 |
]
|
| 22 |
|
| 23 |
+
examples = [
|
| 24 |
+
['./aeroplane.png'],
|
| 25 |
+
['./horse.png']
|
| 26 |
+
]
|
| 27 |
+
|
| 28 |
IMG_SIZE = 72
|
| 29 |
|
| 30 |
def infer(input_image):
|
|
|
|
| 47 |
|
| 48 |
input = gr.inputs.Image(shape=(IMG_SIZE, IMG_SIZE))
|
| 49 |
output = [gr.outputs.Label()]
|
|
|
|
| 50 |
title = "Image Classification using "
|
| 51 |
description = "Upload an image or select from examples to classify it.<br>The allowed classes are - Airplane, Automobile, Bird, Cat, Deer, Dog, Frog, Horse, Ship, Truck.<br><p><b>Space author: Vivek Rai</b> <br><b> Keras example author: Sayak Paul </b></p>"
|
| 52 |
|