Update app.py

app.py

@@ -5,6 +5,14 @@ import numpy as np
 import cv2
 from PIL import Image
 
+# Initialize the model
+config = DetrConfig.from_pretrained("facebook/detr-resnet-50")
+model = DetrForObjectDetection.from_pretrained("facebook/detr-resnet-50", config=config)
+image_processor = DetrImageProcessor.from_pretrained("facebook/detr-resnet-50")
+
+# Initialize the pipeline
+od_pipe = pipeline(task='object-detection', model=model, image_processor=image_processor)
+
 def draw_detections(image, detections):
     # Convert PIL image to a numpy array
     np_image = np.array(image)