initial commit

app.py

@@ -39,6 +39,10 @@ trapezoid_pts = np.array([[250, 150], [400, 150], [450, 300], [200, 300]], np.in
 # Create temp directory for uploads if it doesn't exist
 os.makedirs("temp", exist_ok=True)
 
+"""Load the YOLO model when the app starts."""
+model = YOLO(model_name)
+print(f"Model {model_name} loaded successfully")
+
 def is_inside_trapezoid(box, trapezoid_pts):
     """Check if the center of a detected object is inside the trapezoidal area."""
     x1, y1, x2, y2 = box
@@ -108,12 +112,6 @@ def encode_image_to_base64(image):
     _, buffer = cv2.imencode('.jpg', image)
     return base64.b64encode(buffer).decode('utf-8')
 
-@app.on_event("startup")
-async def startup_event():
-    """Load the YOLO model when the app starts."""
-    global model
-    model = YOLO(model_name)
-    print(f"Model {model_name} loaded successfully")
 
 @app.get("/")
 async def root():