Update sd/zerogpu_controller.py

sd/zerogpu_controller.py

@@ -34,7 +34,7 @@ SECOND_PIPE=get_pipe(vae=VAE,
 def get_first_result(img, prompt, negative_prompt, 
                      controlnet_scale=0.5, strength=1.0,n_steps=30,eta=1.0):
 
-    substrate, resized_image = sketch_process(img)
+    substrate, resized_image = sketch_process(img["composite"])
     prompt=prompt_preprocess(prompt)
 
     FIRST_PIPE.to('cuda')
@@ -83,6 +83,6 @@ def get_second_result(img, prompt, negative_prompt,
 
 
 def get_help_w_prompt(img):
-    return helper.get_help(img["composite"])#Image.fromarray(img["composite"], "RGB"))
+    return helper.get_help(img["composite"]) #Image.fromarray(img["composite"], "RGB"))
 
 #ui(get_first_result, get_second_result) #controller)