update

README.md

@@ -1,5 +1,5 @@
 ---
-title: Kolors-Controlnet
+title: Kolors Controlnet
 emoji: 🏞️
 colorFrom: purple
 colorTo: green

app.py

@@ -88,8 +88,8 @@ model_dwpose = DWposeDetector()
 def process_dwpose_condition(image, res=1024):
     h,w,_ = image.shape
     img = resize_image(HWC3(image), res)
-    out_res, out_img = model_dwpose(img) 
-    result = HWC3( out_img )
+    out_res, out_img = model_dwpose(image) 
+    result = HWC3(out_img)
     result = cv2.resize( result, (w,h) )
     return Image.fromarray(result)
 

assets/title.md

@@ -1,7 +1,7 @@
 <div style="display: flex; justify-content: center; align-items: center; text-align: center;">
   <div>
-    <h1>Kolors-Controlnet</h1>
-    <span>Two ControlNet based on Kolors-Basemodel: Canny and Depth</span>
+    <h1>Kolors Controlnet</h1>
+    <span>Three ControlNet based on Kolors-Basemodel: Canny, Depth and Pose</span>
     <br>
     <div style="display: flex; justify-content: center; align-items: center; text-align: center;">
         <a href="https://github.com/Kwai-Kolors/Kolors/tree/master/controlnet"><img src="https://img.shields.io/static/v1?label=Kolors Code&message=Github&color=blue&logo=github-pages"></a> &ensp;