fix bugs

main.py

@@ -2,7 +2,8 @@ import logging
 import math
 import os
 from typing import Any, Dict, List, Optional, Tuple, Union
-from diffusers.models.controlnet import ControlNetConditioningEmbedding
+#from diffusers.models.controlnet import ControlNetConditioningEmbedding
+from diffusers.models.controlnets.controlnet import ControlNetConditioningEmbedding
 import torch
 from torch import nn
 import torch.nn.functional as F
@@ -39,7 +40,7 @@ from diffusers import UniPCMultistepScheduler
 from src.models.stage2_inpaint_unet_2d_condition import Stage2_InapintUNet2DConditionModel
 
 from torchvision import transforms
-from diffusers.models.controlnet import ControlNetConditioningEmbedding
+#from diffusers.models.controlnet import ControlNetConditioningEmbedding
 from transformers import CLIPImageProcessor
 from transformers import Dinov2Model
 from diffusers import AutoencoderKL, DDPMScheduler, UNet2DConditionModel,ControlNetModel,DDIMScheduler
@@ -49,6 +50,7 @@ from src.pipelines.PCDMs_pipeline import PCDMsPipeline
 
 import spaces
 from libs.easy_dwpose import DWposeDetector
+from libs.easy_dwpose.draw import draw_openpose
 from PIL import Image
 import cv2
 import os
@@ -1160,6 +1162,7 @@ def run_generate_frame(images, target_poses, train_steps=100, inference_steps=10
     is_app=True
 
     print(target_poses)
+    target_poses = [draw_openpose(pose, height=img_height, width=img_width, include_hands=True, include_face=False) for pose in target_poses]
     
     dwpose, rembg_session, pcdms_model, noise_scheduler, image_encoder_p, image_encoder_g, vae, unet = load_models()
     

requirements.txt

@@ -5,3 +5,4 @@ accelerate
 gradio
 rembg[cpu]
 spaces
+matplotlib