Update app.py

app.py

@@ -10,7 +10,7 @@ from PIL import Image
 from diffusers import QwenImageControlNetPipeline, QwenImageControlNetModel
 
 # --- Preprocessor Imports ---
-from controlnet_aux import OpenposeDetector, AnylineDetector
+from controlnet_aux import OpenposeDetector #, AnylineDetector
 from depth_anything_v2.dpt import DepthAnythingV2
 
 # --- Prompt Enhancement Imports ---
@@ -98,7 +98,7 @@ depth_anything = depth_anything.to(device).eval()
 # Load Pose and Soft Edge Detectors
 print("Loading other detectors...")
 openpose_detector = OpenposeDetector.from_pretrained("lllyasviel/Annotators")
-anyline_detector = AnylineDetector.from_pretrained("lllyasviel/Annotators", filename="anyline.pth").to(device)
+#anyline_detector = AnylineDetector.from_pretrained("lllyasviel/Annotators", filename="anyline.pth").to(device)
 
 print("All models loaded.")
 
@@ -106,8 +106,8 @@ def get_control_image(input_image, control_mode):
     """A master function to select and run the correct preprocessor."""
     if control_mode == "Canny":
         return extract_canny(input_image)
-    elif control_mode == "Soft Edge":
-        return anyline_detector(input_image, to_pil=True)
+    #elif control_mode == "Soft Edge":
+    #    return anyline_detector(input_image, to_pil=True)
     elif control_mode == "Depth":
         image_np = np.array(input_image)
         with torch.no_grad():