Spaces:
Runtime error
Runtime error
fix(gradio): fix type in input'
Browse files
app.py
CHANGED
@@ -23,10 +23,11 @@ def predict(input_img) -> tuple[np.ndarray | PIL.Image.Image | str, list[tuple[n
|
|
23 |
|
24 |
gradio_app = gr.Interface(
|
25 |
predict,
|
26 |
-
inputs=gr.Image(label="
|
27 |
# output displays the image with the bounding boxes
|
28 |
outputs=gr.AnnotatedImage(),
|
29 |
title="Detect Watermark in Images",
|
|
|
30 |
examples=[
|
31 |
os.path.join(os.path.dirname(__file__), "samples/example_text1.jpg"),
|
32 |
os.path.join(os.path.dirname(__file__), "samples/example_text2.jpg"),
|
|
|
23 |
|
24 |
gradio_app = gr.Interface(
|
25 |
predict,
|
26 |
+
inputs=gr.Image(label="Upload your watermaked image", sources=['upload'], type="pil"),
|
27 |
# output displays the image with the bounding boxes
|
28 |
outputs=gr.AnnotatedImage(),
|
29 |
title="Detect Watermark in Images",
|
30 |
+
description="This demo use a YoloV8 Nano model from Ultralytics, fine-tuned on the PITA Dataset for watermarked images",
|
31 |
examples=[
|
32 |
os.path.join(os.path.dirname(__file__), "samples/example_text1.jpg"),
|
33 |
os.path.join(os.path.dirname(__file__), "samples/example_text2.jpg"),
|