Update app.py

app.py

@@ -153,11 +153,18 @@ def run_image(runner,
     return image
 
 
+
+
+
 def demo(runner, vision_encoder, vision_processor, padding_embed):
     with gr.Blocks(title="YOLO-World") as demo:
         with gr.Row():
-            gr.Markdown('<h1><center>YOLO-World: Real-Time Open-Vocabulary '
-                        'Object Detector</center></h1>')
+            gr.Markdown('<h1><center>YOLO-World-Image: YOLO-World with Image Prompts</center></h1>')
+            gr.Markdown('<center> This demo is actively under construction and the YOLO-World-Image is an initial version!')
+            gr.Markdown("""<center>
+            [![project page](https://img.shields.io/badge/Project-Page-green)](https://yoloworld.cc)
+            [![arxiv paper](https://img.shields.io/badge/arXiv-Paper-red)](https://arxiv.org/abs/2401.17270)
+            [![demo](https://img.shields.io/badge/🤗HugginngFace-Spaces-orange)](https://huggingface.co/spaces/stevengrove/YOLO-World)</center>""")
         with gr.Row():
             image = gr.Image(type='pil', label='input image')
             output_image = gr.Image(type='pil', label='output image')
@@ -215,22 +222,10 @@ def demo(runner, vision_encoder, vision_processor, padding_embed):
                     exp_image_dir + "1.png", exp_prompt_dir + "1.png", "",
                     "padding", 0.2, 0.1, 100
                 ],
-                [
-                    exp_image_dir + "2.png", exp_prompt_dir + "2.png", "",
-                    "padding", 0.0, 0.1, 200
-                ],
                 [
                     exp_image_dir + "3.png", exp_prompt_dir + "3.png", "",
                     "padding", 0.3, 0.5, 100
                 ],
-                [
-                    exp_image_dir + "4.png", exp_prompt_dir + "4.png", "",
-                    "padding", 0.01, 0.1, 200
-                ],
-                [
-                    exp_image_dir + "5.png", exp_prompt_dir + "5.png", "",
-                    "none", 0.3, 0.5, 100
-                ],
             ],
             inputs=[
                 image, prompt_image, input_text, add_padding, score_thr,