aiunivers commited on
Commit
3ac75c6
·
verified ·
1 Parent(s): 899482e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -9
app.py CHANGED
@@ -1,19 +1,16 @@
1
  from PIL import Image
2
  import gradio as gr
3
-
4
- # ControlNet OpenPose from the exact version you used
5
  from controlnet_aux.openpose import OpenposeDetector
6
 
7
- # Load OpenPose detector (hands + face included by default in this version)
8
  openpose = OpenposeDetector.from_pretrained("lllyasviel/ControlNet")
9
 
10
- # Function to generate pose
11
- def generate_pose(image: Image.Image, use_openpose: bool = True):
12
- image = image.convert("RGB")
13
  if use_openpose:
14
- result = openpose(image)
15
  else:
16
- result = image # fallback, no depth
17
  if not isinstance(result, Image.Image):
18
  result = Image.fromarray(result)
19
  return result
@@ -27,7 +24,7 @@ demo = gr.Interface(
27
  ],
28
  outputs=gr.Image(type="pil", label="Pose Output"),
29
  title="OpenPose Pose Generator",
30
- description="Upload an image and generate a pose map (body + face + hands)."
31
  )
32
 
33
  if __name__ == "__main__":
 
1
  from PIL import Image
2
  import gradio as gr
 
 
3
  from controlnet_aux.openpose import OpenposeDetector
4
 
5
+ # Load OpenPose detector
6
  openpose = OpenposeDetector.from_pretrained("lllyasviel/ControlNet")
7
 
8
+ def generate_pose(image, use_openpose=True):
9
+ img = image.convert("RGB")
 
10
  if use_openpose:
11
+ result = openpose(img)
12
  else:
13
+ result = img
14
  if not isinstance(result, Image.Image):
15
  result = Image.fromarray(result)
16
  return result
 
24
  ],
25
  outputs=gr.Image(type="pil", label="Pose Output"),
26
  title="OpenPose Pose Generator",
27
+ description="Generate full body pose including face and hands."
28
  )
29
 
30
  if __name__ == "__main__":