Update app.py

app.py

@@ -1,5 +1,4 @@
 from fastapi import FastAPI, UploadFile, File
-from ultralytics import YOLO
 from transformers import pipeline
 from PIL import Image
 import io
@@ -8,16 +7,22 @@ import io
 app = FastAPI()
 
 # Load the YOLOv5 model for object detection
-model = YOLO("yolov8n.pt")
+from transformers import pipeline
+from PIL import Image
+
+pipe = pipeline("object-detection", model="facebook/detr-resnet-50")
+
+
+bounding_boxes = pipe(image)
 
 @app.post("/detect_objects/")
 async def detect_objects(file: UploadFile = File(...)):
     # Read image from uploaded file
     image_bytes = await file.read()
-    image = Image.open(io.BytesIO(image_bytes))
+    image = Image.open(io.BytesIO(image_bytes)).convert("RGB")
     
     # Perform object detection
-    results = model(image)
+    results = pipe(image)
     
     # Extract bounding box coordinates and labels
     bounding_boxes = []