trying new subprocess thing

app.py

@@ -35,8 +35,6 @@ 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
-accelerator = accelerate.Accelerator()
 
 CT_HOME = '.'
 
@@ -114,11 +112,6 @@ class Args:
 @spaces.GPU
 def process_image(image, repaint, busyness):
 
-    print("CUDA check:")
-    print("   zero.device = ",zero.device)
-    device = accelerator.device
-    print('   accelerator.device:', device, flush=True)
-
     # get image ready and execute sampler
     print("image = ",image)
     image = image['composite']
@@ -140,17 +133,24 @@ 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) 
-    # Commenting out the following since HF ZeroGPU doesn't seem to work with subprocesses. 
-    # 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, seed_scale=0.0, repaint=repaint)
-    print(" Now calling zero_wrapper with args = ",args,"\n")
-    zero_wrapper(args, accelerator, device)
+
+    #  HF ZeroGPU+Gradio doesn't seem to work with subprocesses.
+    use_subprocess = True 
+    if use_subprocess:
+        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 subprocess with args = ", args,"\n")
+        return_value = call(args)
+        print("Subprocess finished.  Return value = ", return_value)
+    else:
+        accelerator = accelerate.Accelerator()
+        device = accelerator.device
+        print("Accelerator device = ", device)
+        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, seed_scale=0.0, 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'