now calling zero_wrapper directly

app.py

@@ -29,6 +29,8 @@ import k_diffusion as K
 import natten
 import accelerate
 
+from sample import zero_wrapper
+
 zero = torch.Tensor([0]).cuda()
 print("Zero Device = ",zero.device," <-- this probably says cpu") # <-- 'cpu' 🤔
 
@@ -96,6 +98,12 @@ def grab_dense_gen(init_img,
     print("Grabbing filename = ", dense_filename)
     return dense_filename
 
+# dummy class to make an args-like object
+class Args:
+    def __init__(self, **kwargs):
+        for key, value in kwargs.items():
+            setattr(self, key, value)
+
 
 @spaces.GPU
 def process_image(image, repaint, busyness):
@@ -126,14 +134,16 @@ def process_image(image, repaint, busyness):
     PREFIX = 'gradiodemo'
     # !echo {DEVICES} {CT_HOME} {CKPT} {PREFIX} {masked_img_file}
     print("Reading init image from ", masked_img_file,", repaint = ",repaint) 
-    cmd = f'{sys.executable} {CT_HOME}/sample.py --batch-size {bs} --checkpoint {CKPT} --config {CT_HOME}/configs/config_pop909_256x256_chords.json -n {num} --prefix {PREFIX} --init-image {masked_img_file} --steps=100 --repaint={repaint}'    
-    print("Will run command: ", cmd)
-    args = cmd.split(' ')
-    #call(cmd, shell=True)
-    print("Calling: ", args,"\n")
-    return_value = call(args)
-    print("Return value = ", return_value)
-
+    # cmd = f'{sys.executable} {CT_HOME}/sample.py --batch-size {bs} --checkpoint {CKPT} --config {CT_HOME}/configs/config_pop909_256x256_chords.json -n {num} --prefix {PREFIX} --init-image {masked_img_file} --steps=100 --repaint={repaint}'    
+    # print("Will run command: ", cmd)
+    # args = cmd.split(' ')
+    # #call(cmd, shell=True)
+    # print("Calling: ", args,"\n")
+    # return_value = call(args)
+    # print("Return value = ", return_value)
+    args = Args(batch_size=bs, checkpoint=CKPT, config=f'{CT_HOME}/configs/config_pop909_256x256_chords.json', n=num, prefix=PREFIX, init_image=masked_img_file, steps=100, repaint=repaint)
+    print(" Now calling zero_wrapper with args = ",args,"\n")
+    zero_wrapper(args, accelerator, device)
 
     # find gen'd image and convert to midi piano roll 
     #gen_file = f'{PREFIX}_00000.png'

sample.py

@@ -22,8 +22,6 @@ from pom.v_diffusion import DDPM, LogSchedule, CrashSchedule
 #CHORD_BORDER = 8   # chord border size in pixels
 from pom.chords import CHORD_BORDER, img_batch_to_seq_emb, ChordSeqEncoder
 
-import spaces
-
 # ---- my mangled sampler that includes repaint 
 import torchsde 
 
@@ -516,8 +514,8 @@ def get_init_image_and_mask(args, device):
 
 
 # wrapper compatible with ZeroGPU, callable from outside
-@spaces.GPU
-def zero_wrapper(args, device): 
+#@spaces.GPU
+def zero_wrapper(args, accelerator, device): 
     global init_image, init_mask
 
     config = K.config.load_config(args.config if args.config else args.checkpoint)