22

app.py

@@ -102,18 +102,18 @@ def inference(
     
     # input_image, width, height = scale_image(input_image, image_scale)
     # control_mode='HED'
-
+    processor_id = 'canny'
     if control_mode == 'HED':
-        processor = Processor('softedge_hed')
+        processor_id = 'softedge_hed'
     if control_mode =='Midas':
-        processor = Processor('depth_midas')
+        processor_id = 'depth_midas'
     if control_mode =='MLSD':
-        processor = Processor('mlsd')
+        processor_id = 'mlsd'
     if control_mode =='Pose':
-        processor = Processor('openpose_full')
-    else:
-        processor = Processor('canny')
+        processor_id = 'openpose_full'
 
+    print(f"DEBUG: processor_id={processor_id}")
+    processor = Processor(processor_id)
 
     # Width must be divisible by 16
     control_image, width, height = rescale_image(input_image, image_scale, 16)
@@ -122,7 +122,7 @@ def inference(
     print("DEBUG: processor running")
     control_image = processor(control_image, to_pil=True)
     control_image = control_image.resize((width, height))
-    
+
     print("DEBUG: control_image_torch")
     control_image_torch = get_image_latent(control_image, sample_size=[height, width])[:, :, 0]