Update app.py

app.py

@@ -11,7 +11,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 ---
@@ -99,7 +99,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.")
 
@@ -107,8 +107,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():
@@ -118,10 +118,6 @@ def get_control_image(input_image, control_mode):
         return Image.fromarray(depth).convert('RGB')
     elif control_mode == "Pose":
         return openpose_detector(input_image, hand_and_face=True)
-    elif control_mode == "Recolor":
-        return convert_to_grayscale(input_image)
-    elif control_mode == "Tile":
-        return tile_image(input_image, 16)
     else:
         raise ValueError(f"Unknown control mode: {control_mode}")
 
@@ -185,7 +181,7 @@ with gr.Blocks(css="footer {display: none !important;}") as demo:
             input_image = gr.Image(type="pil", label="Input Image", height=512)
             prompt = gr.Textbox(label="Prompt", placeholder="A detailed description of the desired image...")
             conditioning = gr.Radio(
-                choices=["Canny", "Soft Edge", "Depth", "Pose", "Recolor", "Tile"],
+                choices=["Canny", "Soft Edge", "Depth", "Pose"],
                 value="Pose",
                 label="Conditioning Type"
             )