enable zerogpu

app.py

@@ -256,9 +256,6 @@ hands = mp_hands.Hands(
 
 @spaces.GPU(duration=120)
 def get_ref_anno(ref):
-    print(torch.cuda.is_available())
-    print(torch.cuda.device_count())
-    print(torch.cuda.get_device_name(0))
     if ref is None:
         return (
             None,
@@ -362,6 +359,7 @@ def get_ref_anno(ref):
         target_size=opts.image_size,
         latent_size=opts.latent_size,
     )
+    print("ready to go to autoencoder")
     latent = opts.latent_scaling_factor * autoencoder.encode(image).sample()
     if not REF_POSE_MASK:
         heatmaps = torch.zeros_like(heatmaps)
@@ -1555,4 +1553,4 @@ with gr.Blocks(css=custom_css) as demo:
 # _, _, shared_url = demo.queue().launch(
 #     share=True, server_name="0.0.0.0", server_port=7739
 # )
-demo.launch()
+demo.launch(share=True)