Bhaskar Saranga commited on
Commit
9ff44a7
1 Parent(s): d2228f2

Fixed GPU issue

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -126,7 +126,7 @@ def detectv8(img,model,device,iou_threshold=0.45,confidence_threshold=0.25):
126
  fps_inference = 1/(time.time()-start)
127
 
128
  if torch.cuda.is_available():
129
- boxes= results[0].boxes.cpu().detach().numpy()
130
  else:
131
  boxes=results[0].boxes.numpy()
132
  for bbox in boxes:
 
126
  fps_inference = 1/(time.time()-start)
127
 
128
  if torch.cuda.is_available():
129
+ boxes= results[0].boxes.cpu().numpy()
130
  else:
131
  boxes=results[0].boxes.numpy()
132
  for bbox in boxes: