fixed zero def

app.py

@@ -35,6 +35,8 @@ from sample import zero_wrapper
 from pom.pianoroll import regroup_lines, img_file_2_midi_file, square_to_rect, rect_to_square
 from pom.square_to_rect import square_to_rect
 
+zero = torch.Tensor([0]).cuda()  # used for gpu checking later
+
 CT_HOME = '.'
 
 @spaces.GPU
@@ -112,7 +114,6 @@ class Args:
 def process_image(image, repaint, busyness):
 
     print("CUDA check:")
-    zero = torch.Tensor([0]).cuda()
     print("   zero.device = ",zero.device)
     accelerator = accelerate.Accelerator()
     device = accelerator.device