Update app.py

app.py

@@ -4,12 +4,14 @@ from transformers import DetrForObjectDetection, DetrImageProcessor, AutoModel
 import supervision as sv
 from supervision.detection.annotate import BoxAnnotator
 from supervision.utils.notebook import plot_image
+import cv2
 
 og_model = 'facebook/detr-resnet-50'
 image_processor =  DetrImageProcessor.from_pretrained(og_model)
 model = AutoModel.from_pretrained("taroii/notfinetuned-detr-50")
 
-def predict(image):
+def predict(image_path):
+    image = cv2.imread(image_path)
     with torch.no_grad():
         # load image and predict
         inputs = image_processor(images=image, return_tensors='pt')