Spaces:

Chaerin5
/

FoundHand

Running on Zero

Chaerin5 commited on Dec 29, 2024

Commit

a366fc8

1 Parent(s): a4e93b7

enable zerogpu

Files changed (1) hide show

app.py CHANGED Viewed

@@ -341,7 +341,7 @@ def get_ref_anno(ref):
             Normalize(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5], inplace=True),
         ]
     )
-    image = image_transform(img).to(device)
     kpts_valid = check_keypoints_validity(keypts, opts.image_size)
     heatmaps = torch.tensor(
         keypoint_heatmap(

             Normalize(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5], inplace=True),
         ]
     )
+    image = image_transform(img)
     kpts_valid = check_keypoints_validity(keypts, opts.image_size)
     heatmaps = torch.tensor(
         keypoint_heatmap(