22

app.py

@@ -10,7 +10,6 @@ from utils.image_utils import get_image_latent, scale_image
 from utils.prompt_utils import polish_prompt
 # from controlnet_aux import HEDdetector, MLSDdetector, OpenposeDetector, CannyDetector, MidasDetector
 from controlnet_aux.processor import Processor
-# from videox_fun.utils.utils import get_image_latent
 
 
 # MODEL_REPO = "Tongyi-MAI/Z-Image-Turbo"
@@ -95,6 +94,7 @@ def inference(
     progress=gr.Progress(track_tqdm=True),
 ):
     # process image
+    print("DEBUG: process image")
     if input_image is None:
         print("Error: input_image is empty.")
         return None
@@ -113,10 +113,13 @@ def inference(
     else:
         processor = Processor('canny')
 
-    control_image, width, height = scale_image(input_image, image_scale, 64)
+    control_image, width, height = scale_image(input_image, image_scale, 8)
     control_image = control_image.resize((512, 512))
+
+    print("DEBUG: processor running")
     control_image = processor(control_image, to_pil=True)
 
+    print("DEBUG: control_image_torch")
     control_image_torch = get_image_latent(control_image, sample_size=[height, width])[:, :, 0]
 
     # generation
@@ -227,8 +230,8 @@ with gr.Blocks(css=css) as demo:
                         value=8,
                     )
         gr.Examples(examples=examples, inputs=[input_image, prompt])
-        
         gr.HTML(read_file("static/footer.html"))
+
     run_button.click(
         fn=prepare,
         inputs=prompt,

static/header.html

@@ -8,6 +8,6 @@
             <br>
             Supports multiple control conditions - including Canny, HED, Depth, Pose and MLSD.
             <br>
-            If you like my spaces, please support me by visiting <a href="https://aisudo.com/" target="_blank">AiSudo</a> for more image generation 😊
+            If you like my spaces, please support me by visiting <a href="https://aisudo.com/" target="_blank">AiSudo</a> 😊
     </div>
 </div>