Update ip_adapter_openpose.py

ip_adapter_openpose.py

@@ -3,16 +3,21 @@ import torch
 
 from openpose import get_openpose, init as init_openpose
 from adapter_model import MODEL
+from img2txt import derive_caption,init as init_img2txt
 
 init_openpose()
+init_img2txt()
+
 ip_model = MODEL("pose")
 
 
 def generate(img_human: Image, img_clothes: Image):
 
     img_openpose = get_openpose(img_human)
-
+    prompt = derive_caption(img_clothes)
+    
     img_openpose_gen = ip_model.model.generate(
+        prompt=prompt,
         pil_image=img_clothes,
         image=img_openpose,
         width=512,