qfisch commited on
Commit
05629a1
·
1 Parent(s): 84f5b3e

fix(gradio): fix type in input'

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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="Update 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
  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"),