Update app.py

app.py

@@ -13,7 +13,7 @@ from fastapi.responses import HTMLResponse
 app = FastAPI()
 
 # Load YOLO model and captioning model
-yolo_model = YOLO('best.pt').to('cpu')  # Use 'cuda' if available for GPU
+yolo_model = YOLO('best.pt').to('cuda')  # Use 'cuda' if available for GPU
 
 # Initialize captioning model (Florence)
 processor = AutoProcessor.from_pretrained("microsoft/Florence-2-base", trust_remote_code=True)