cuda validation for v8 predictions

app.py

@@ -125,7 +125,10 @@ def detectv8(img,model,device,iou_threshold=0.45,confidence_threshold=0.25):
     results= model.predict(img,conf=confidence_threshold, iou=iou_threshold)
     fps_inference = 1/(time.time()-start)
     
-    boxes=results[0].boxes.numpy()
+    if torch.cuda.is_available():
+        boxes= results[0].boxes.cpu().detach().numpy()
+    else:
+        boxes=results[0].boxes.numpy()
     for bbox in boxes:
         #print(f'{colors[names[int(bbox.cls[0])]]}')
         label = f'{names[int(bbox.cls[0])]} {bbox.conf[0]:.2f}'