Update app.py

app.py

@@ -79,8 +79,11 @@ def detect_objects(model_name,url_input,image_input,threshold):
     
     return viz_img   
         
-#examples=[['facebook/detr-resnet-50','https://media-cldnry.s-nbcnews.com/image/upload/t_fit-1500w,f_auto,q_auto:best/newscms/2020_14/3290756/200331-wall-street-ew-#343p.jpg',,0.7]
+def set_example_image(example: list) -> dict:
+    return gr.Image.update(value=example[0])
 
+def set_example_url(example: list) -> dict:
+    return gr.Textbox.update(value=example[0])
 
 
 title = """<h1 id="title">Object Detection App with DETR and YOLOS</h1>"""
@@ -96,7 +99,12 @@ Links to HuggingFace Models:
 
 models = ["facebook/detr-resnet-50","facebook/detr-resnet-101",'hustvl/yolos-small']
 
-demo = gr.Blocks()
+css = '''
+h1#title {
+  text-align: center;
+}
+'''
+demo = gr.Blocks(css=css)
 
 with demo:
     gr.Markdown(title)
@@ -133,6 +141,7 @@ with demo:
     
     url_but.click(detect_objects,inputs=[options,url_input,img_input,slider_input],outputs=img_output_from_url,queue=True)
     img_but.click(detect_objects,inputs=[options,url_input,img_input,slider_input],outputs=img_output_from_upload,queue=True)
-    
+    example_images.click(fn=set_example_image,inputs=[example_images],outputs=[image_input])
+    example_url.click(fn=set_example_url,inputs=[example_url],outputs=[url_input])
     
 demo.launch(enable_queue=True)