Spaces:
Build error
Build error
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()
|