Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -6,7 +6,7 @@ model = tf.keras.models.load_model("hf://JaviSwift/cifar10_simple")
|
|
6 |
|
7 |
def predict_image(img):
|
8 |
"""
|
9 |
-
|
10 |
"""
|
11 |
|
12 |
img = tf.image.resize(img, (32, 32))
|
@@ -24,10 +24,10 @@ def predict_image(img):
|
|
24 |
|
25 |
iface = gr.Interface(
|
26 |
fn=predict_image,
|
27 |
-
inputs=gr.Image(label="
|
28 |
-
outputs=gr.Label(label="
|
29 |
-
title="
|
30 |
-
description="
|
31 |
)
|
32 |
|
33 |
iface.launch()
|
|
|
6 |
|
7 |
def predict_image(img):
|
8 |
"""
|
9 |
+
Makes a prediction of the image descripton
|
10 |
"""
|
11 |
|
12 |
img = tf.image.resize(img, (32, 32))
|
|
|
24 |
|
25 |
iface = gr.Interface(
|
26 |
fn=predict_image,
|
27 |
+
inputs=gr.Image(label="Upload an image"),
|
28 |
+
outputs=gr.Label(label="Result"),
|
29 |
+
title="Image description predictor",
|
30 |
+
description="Upload an image and it will make a description of the object"
|
31 |
)
|
32 |
|
33 |
iface.launch()
|