Update demo/model.py

demo/model.py

@@ -384,7 +384,7 @@ class Model_all:
                 pose_link_color=self.pose_link_color,
                 radius=2,
                 thickness=2)
-            im_keypose_out = im_keypose_out.astype(np.uint8)[:, :, ::-1]
+            im_keypose_out = im_keypose_out.astype(np.uint8)
 
         # extract condition features
         c = self.base_model.get_learned_conditioning([prompt + ', ' + pos_prompt])
@@ -419,7 +419,7 @@ class Model_all:
         x_samples_ddim = 255. * x_samples_ddim
         x_samples_ddim = x_samples_ddim.astype(np.uint8)
 
-        return [im_depth_out, im_keypose_out, x_samples_ddim]
+        return [im_depth_out, im_keypose_out[:, :, ::-1], x_samples_ddim]
 
     @torch.no_grad()
     def process_seg(self, input_img, type_in, prompt, neg_prompt, pos_prompt, fix_sample, scale,