:bug: trying to solve runtime error in :hugging: spaces; detach!

.gitignore

@@ -0,0 +1,2 @@
+*.jpg
+*.pt

app.py

@@ -43,7 +43,7 @@ def detect(pil_img,names):
     img /= 255.0  # 0 - 255 to 0.0 - 1.0
 
     # Inference
-    pred = model(img.unsqueeze(0).permute(0,3,1,2), augment=False)[0]
+    pred = model(img.unsqueeze(0).permute(0,3,1,2), augment=False)[0].detach()
 
     # Apply NMS
     pred = non_max_suppression(pred, conf_thres, iou_thres, classes=None, agnostic=False)