fix bug

demo/model.py

@@ -378,6 +378,7 @@ class Model_all:
                 pose_link_color=self.pose_link_color,
                 radius=2,
                 thickness=2)
+            im_keypose_out = cv2.resize(im_keypose_out, (512, 512))
 
         # extract condition features
         c = self.base_model.get_learned_conditioning([prompt + ', ' + pos_prompt])