Update app.py

app.py

@@ -1,6 +1,5 @@
 import matplotlib.pyplot as plt
 import numpy as np
-from six import BytesIO
 from PIL import Image
 import tensorflow as tf
 from object_detection.utils import label_map_util
@@ -10,7 +9,8 @@ import tarfile
 import wget 
 import gradio as gr
 from huggingface_hub import snapshot_download
-import os 
+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)
@@ -20,13 +20,7 @@ def pil_image_as_numpy_array(pilimg):
     img_array = tf.keras.utils.img_to_array(pilimg)
     # img_array = np.expand_dims(img_array, axis=0)
     return img_array
-    
-def load_image_into_numpy_array(path):
-                                    
-    image = None
-    image_data = tf.io.gfile.GFile(path, 'rb').read()
-    image = Image.open(BytesIO(image_data))
-    return pil_image_as_numpy_array(image)            
+        
 
 def load_model():
     model_dir = 'saved_model'