22

app.py

@@ -99,7 +99,7 @@ def inference(
         print("Error: input_image is empty.")
         return None
     
-    input_image, width, height = scale_image(input_image, image_scale)
+    # input_image, width, height = scale_image(input_image, image_scale)
     control_mode='HED'
 
     if control_mode == 'HED':
@@ -113,7 +113,8 @@ def inference(
     else:
         processor = Processor('canny')
 
-    control_image = scale_image(input_image, image_scale, 64)
+    control_image, width, height = scale_image(input_image, image_scale, 64)
+    control_image = control_image.resize(512, 512)
     control_image = processor(control_image, to_pil=True)
 
     control_image_torch = get_image_latent(control_image, sample_size=[height, width])[:, :, 0]