Update app.py

app.py

@@ -1,4 +1,5 @@
 from fastapi import FastAPI, UploadFile, File
+from ultralytics import YOLO
 from transformers import pipeline
 from PIL import Image
 import io
@@ -7,7 +8,7 @@ import io
 app = FastAPI()
 
 # Load the YOLOv5 model for object detection
-detector = pipeline('object-detection', model='Ultralytics/YOLOv8')
+model = YOLO("yolov8n.pt")
 
 @app.post("/detect_objects/")
 async def detect_objects(file: UploadFile = File(...)):
@@ -16,7 +17,7 @@ async def detect_objects(file: UploadFile = File(...)):
     image = Image.open(io.BytesIO(image_bytes))
     
     # Perform object detection
-    results = detector(image)
+    results = model(image)
     
     # Extract bounding box coordinates and labels
     bounding_boxes = []