Update app.py

app.py

@@ -145,6 +145,9 @@ class LitYolo(LightningModule):
 # gradio
 with gr.Blocks() as demo:
 
+    # example images
+    examples = ["example_images/000041.jpg","example_images/000042.jpg","example_images/000043.jpg","example_images/000044.jpg","example_images/000045.jpg"]
+    
     # colors for the bboxes
     cmap = plt.get_cmap("tab20b")
     class_labels = config.PASCAL_CLASSES
@@ -204,7 +207,13 @@ with gr.Blocks() as demo:
         sections.append(((upper_left_x,upper_left_y,lower_right_x,lower_right_y), class_labels[int(class_pred)]))
       return (np.array(pr_input_img.squeeze(0).permute(1,2,0)),sections)
 
-    section_btn.click(yolo3_inference, img_input, img_output)
+    section_btn.click(yolo3_inference, inputs=[img_input], outputs=[img_output])
+
+    gr.Markdown("## Some Examples")
+    gr.Examples(examples=examples,
+                             inputs =img_input,
+                             outputs=img_output,
+                             fn=yolo3_inference, cache_examples=False)
 
 if __name__ == "__main__":
     demo.launch()