removed a few decorators where they may not be needed

app.py

@@ -38,7 +38,6 @@ from pom.square_to_rect import square_to_rect
 
 CT_HOME = '.'
 
-@spaces.GPU
 def infer_mask_from_init_img(img, mask_with='grey'):
     "note, this works whether image is normalized on 0..1 or -1..1, but not 0..255"
     assert mask_with in ['blue','white','grey']
@@ -62,7 +61,6 @@ def infer_mask_from_init_img(img, mask_with='grey'):
     return mask*1.0
 
 
-@spaces.GPU
 def count_notes_in_mask(img, mask):
     "counts the number of new notes in the mask"
     img_t = ToTensor()(img)
@@ -70,7 +68,6 @@ def count_notes_in_mask(img, mask):
     return new_notes.item()
 
 
-@spaces.GPU
 def grab_dense_gen(init_img, 
                 PREFIX, 
                 num_to_gen=64, 

sample.py

@@ -513,10 +513,10 @@ def get_init_image_and_mask(args, device):
 
 
 
-# wrapper compatible with ZeroGPU, callable from outside
-#@spaces.GPU
+# wrapper compatible with ZeroGPU+Gradio, callable from outside
 def zero_wrapper(args, accelerator, device): 
     global init_image, init_mask
+    print("zero_wrapper: Using device:", device, flush=True)
 
     config = K.config.load_config(args.config if args.config else args.checkpoint)
     model_config = config['model']