Update app.py

app.py

@@ -11,6 +11,7 @@ import wget
 import gradio as gr
 from huggingface_hub import snapshot_download
 import os 
+import cv2
 
 PATH_TO_LABELS = 'data/label_map.pbtxt'   
 category_index = label_map_util.create_category_index_from_labelmap(PATH_TO_LABELS, use_display_name=True)
@@ -130,11 +131,17 @@ detection_model = load_model()
 # predicted_img.save('predicted.jpg')
 Image_tab = gr.Interface(fn=predict,
              inputs=gr.Image(type="pil"),
-             outputs=gr.Image(type="pil")
+             outputs=gr.Image(type="pil"),
+             examples=[["SampleImage1.jpg"],["SampleImage2.jpg"],["SampleImage3.jpg"],["SampleImage4.jpg"],["SampleImage5.jpg"],["SampleImage6.jpg"]],
+            title="This is the object detection model for Durian and Pineapple images",
+            description="Using ssd_mobilenet_v2_320x320_coco17_tpu-8 to detect Durina and Pineapple"
              )
 Video_tab = gr.Interface(fn=predict_video,
              inputs=gr.File(label="Upload Video"),
-             outputs=gr.File(label="Detected Video")
+             outputs=gr.File(label="Detected Video"),
+             examples=[["SampleVideo1.mp4"],["SampleVideo2.mp4"]],
+            title="This is the object detection model for Durian and Pineapple videos",
+            description="Using ssd_mobilenet_v2_320x320_coco17_tpu-8 to detect Durina and Pineapple"                         
              )
 
 gr.TabbedInterface([Image_tab, Video_tab], ["Image", "Video"]).launch(share=True)