Spaces:
Runtime error
Runtime error
Bhaskar Saranga
commited on
Commit
·
9ff44a7
1
Parent(s):
d2228f2
Fixed GPU issue
Browse files
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().
|
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:
|