enable zerogpu

app.py

@@ -340,7 +340,7 @@ def get_ref_anno(ref):
     )
     image = image_transform(img)
     image = image.numpy()
-    image = torch.tensor(image)
+    # image = torch.tensor(image)
     kpts_valid = check_keypoints_validity(keypts, opts.image_size)
     heatmaps = torch.tensor(
         keypoint_heatmap(