Better demo title + error management

app.py

@@ -9,6 +9,7 @@ import glob
 import gradio as gr
 from huggingface_hub import get_token
 from utils import (
+    check_image,
     load_image_from_url,
     load_badges,
     FlaggedCounter,
@@ -18,10 +19,11 @@ from model_yolov5 import load_model, inference
 
 
 TITLE = """
-<h1> RGB Detection Demo </h1>
+<h1> 🌊 SEA.AI's Machine Vision Showcase ✨ </h1>
 <p align="center">
-Give it a try! Upload an image or enter a URL to an image and click
-<code>Submit</code>.
+Ahoy! Explore our object detection technology!
+Upload a maritime scene image or use a URL, and click
+<code>Submit</code> to see the results.
 </p>
 """
 
@@ -57,7 +59,9 @@ with gr.Blocks(theme=theme, css=css) as demo:
 
     with gr.Row():
         with gr.Column():
-            img_input = gr.Image(label="input", interactive=True)
+            img_input = gr.Image(
+                label="input", interactive=True, sources=["upload", "clipboard"]
+            )
             img_url = gr.Textbox(
                 lines=1,
                 placeholder="or enter URL to image here",
@@ -68,9 +72,7 @@ with gr.Blocks(theme=theme, css=css) as demo:
                 clear = gr.ClearButton()
                 submit = gr.Button("Submit", variant="primary")
         with gr.Column():
-            img_output = gr.Image(
-                label="output", interactive=False, show_share_button=True
-            )
+            img_output = gr.Image(label="output", interactive=False)
             flag = gr.Button("Flag", visible=False)
             notice = gr.Markdown(value=NOTICE, visible=False)
 
@@ -87,7 +89,7 @@ with gr.Blocks(theme=theme, css=css) as demo:
 
     # event listeners
     img_url.change(load_image_from_url, [img_url], img_input)
-    submit.click(
+    submit.click(check_image, [img_input]).success(
         lambda image: inference(model, image),
         [img_input],
         img_output,

model_yolov5.py

@@ -12,6 +12,7 @@ def load_model(model_path, img_size=640):
     model.img_size = img_size  # add img_size attribute
     return model
 
+
 @spaces.GPU
 def inference(model, image):
     """Run inference on image and return annotated image."""

utils.py

@@ -9,6 +9,12 @@ import gradio as gr
 from huggingface_hub import get_token
 
 
+def check_image(image):
+    """Check image."""
+    if image is None:
+        raise gr.Error("Oops! It looks like you forgot to upload an image.")
+
+
 def load_image_from_url(url):
     """Load image from URL."""
     if not url:  # empty or None