Update app.py

app.py

@@ -1,22 +1,19 @@
 from PIL import Image
 import gradio as gr
 
-# Import OpenPose from controlnet-aux
-try:
-    from controlnet_aux.open_pose import OpenposeDetector
-except ImportError:
-    from controlnet_aux.openpose import OpenposeDetector  # fallback
+# ControlNet OpenPose from the exact version you used
+from controlnet_aux.openpose import OpenposeDetector
 
-# Load OpenPose model with hands and face
-openpose = OpenposeDetector.from_pretrained(
-    "lllyasviel/ControlNet",
-    hand=True,
-    face=True
-)
+# Load OpenPose detector (hands + face included by default in this version)
+openpose = OpenposeDetector.from_pretrained("lllyasviel/ControlNet")
 
-def generate_pose(image: Image.Image):
+# Function to generate pose
+def generate_pose(image: Image.Image, use_openpose: bool = True):
     image = image.convert("RGB")
-    result = openpose(image)
+    if use_openpose:
+        result = openpose(image)
+    else:
+        result = image  # fallback, no depth
     if not isinstance(result, Image.Image):
         result = Image.fromarray(result)
     return result
@@ -24,10 +21,13 @@ def generate_pose(image: Image.Image):
 # Gradio UI
 demo = gr.Interface(
     fn=generate_pose,
-    inputs=gr.Image(type="pil", label="Upload Image"),
+    inputs=[
+        gr.Image(type="pil", label="Upload Image"),
+        gr.Checkbox(value=True, label="Use OpenPose (default: true)"),
+    ],
     outputs=gr.Image(type="pil", label="Pose Output"),
     title="OpenPose Pose Generator",
-    description="Upload an image and generate a pose map with full body, face, and hand keypoints."
+    description="Upload an image and generate a pose map (body + face + hands)."
 )
 
 if __name__ == "__main__":