Chaerin5 commited on
Commit
a0aef17
·
1 Parent(s): e233eca

enable zerogpu

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -256,6 +256,9 @@ hands = mp_hands.Hands(
256
 
257
  @spaces.GPU(duration=120)
258
  def get_ref_anno(ref):
 
 
 
259
  if ref is None:
260
  return (
261
  None,
 
256
 
257
  @spaces.GPU(duration=120)
258
  def get_ref_anno(ref):
259
+ print(torch.cuda.is_available())
260
+ print(torch.cuda.device_count())
261
+ print(torch.cuda.get_device_name(0))
262
  if ref is None:
263
  return (
264
  None,