Chaerin5 commited on
Commit
229f6e0
·
1 Parent(s): 97497a2

enable zerogpu

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -254,7 +254,7 @@ hands = mp_hands.Hands(
254
  min_detection_confidence=0.1,
255
  )
256
 
257
- @spaces.GPU(duration=120)
258
  def get_ref_anno(ref):
259
  print("inside get_ref_anno")
260
  if ref is None:
@@ -270,6 +270,7 @@ def get_ref_anno(ref):
270
  keypts = np.zeros((42, 2))
271
  print("ready to run mediapipe")
272
  if REF_POSE_MASK:
 
273
  mp_pose = hands.process(img)
274
  print("processed mediapipe")
275
  detected = np.array([0, 0])
 
254
  min_detection_confidence=0.1,
255
  )
256
 
257
+ @spaces.GPU(duration=240)
258
  def get_ref_anno(ref):
259
  print("inside get_ref_anno")
260
  if ref is None:
 
270
  keypts = np.zeros((42, 2))
271
  print("ready to run mediapipe")
272
  if REF_POSE_MASK:
273
+ print(f"type(img): {type(img)}, img.shape: {img.shape}, img.dtype: {img.dtype}")
274
  mp_pose = hands.process(img)
275
  print("processed mediapipe")
276
  detected = np.array([0, 0])