Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -150,8 +150,8 @@ def predict(image):
|
|
150 |
# Create the Gradio interface
|
151 |
iface = gr.Interface(
|
152 |
fn=predict,
|
153 |
-
inputs=gr.
|
154 |
-
outputs=gr.
|
155 |
title="Object Detection App",
|
156 |
description="Upload an image, and the YOLO model will detect objects in it."
|
157 |
)
|
|
|
150 |
# Create the Gradio interface
|
151 |
iface = gr.Interface(
|
152 |
fn=predict,
|
153 |
+
inputs=gr.Image(type="pil", label="Upload an Image"), # Upload input as PIL Image
|
154 |
+
outputs=gr.Image(type="pil", label="Predicted Image with Bounding Boxes"), # Output image
|
155 |
title="Object Detection App",
|
156 |
description="Upload an image, and the YOLO model will detect objects in it."
|
157 |
)
|