Update sd/zerogpu_controller.py

sd/zerogpu_controller.py

@@ -2,6 +2,7 @@
 import spaces
 
 from sd.prompt_helper import helper
+from PIL import Image
 
 from sd.utils.utils import *
 from utils.utils import sketch_process, prompt_preprocess
@@ -82,6 +83,6 @@ def get_second_result(img, prompt, negative_prompt,
 
 
 def get_help_w_prompt(img):
-    return helper.get_help(img)
+    return helper.get_help(Image.fromarray(img["composite"]))
 
 #ui(get_first_result, get_second_result) #controller)