Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -13,7 +13,7 @@ model = tf.keras.models.load_model(model_path)
|
|
13 |
|
14 |
|
15 |
# Define prediction function
|
16 |
-
|
17 |
# Save the image to a file-like object
|
18 |
image_bytes = io.BytesIO()
|
19 |
image.save(image_bytes, format="JPEG")
|
|
|
13 |
|
14 |
|
15 |
# Define prediction function
|
16 |
+
def predict_image(image):
|
17 |
# Save the image to a file-like object
|
18 |
image_bytes = io.BytesIO()
|
19 |
image.save(image_bytes, format="JPEG")
|