Spaces:
Runtime error
Runtime error
Johannes
commited on
Commit
·
d2cbc11
1
Parent(s):
fd4d7a2
fix kornia batch problem
Browse files
app.py
CHANGED
@@ -123,7 +123,7 @@ def kornia_detect(img, confidence_threshold, nms_threshold):
|
|
123 |
|
124 |
with torch.no_grad():
|
125 |
dets = face_detection(img)
|
126 |
-
dets = [FaceDetectorResult(o) for o in dets]
|
127 |
|
128 |
# show image
|
129 |
|
|
|
123 |
|
124 |
with torch.no_grad():
|
125 |
dets = face_detection(img)
|
126 |
+
dets = [FaceDetectorResult(o) for o in dets[0]]
|
127 |
|
128 |
# show image
|
129 |
|