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)