'commit'

dsLightRag/Liblib/T3.py

@@ -1,17 +1,15 @@
-import json
-from LiblibUtil import LiblibUtil
-from ControlNetGenerator import ControlNetImageGenerator
+from PuLIDGenerator import PuLIDGenerator
 
 if __name__ == "__main__":
     # 创建生成器实例
-    generator = ControlNetImageGenerator()
-    
+    generator = PuLIDGenerator()
     # 可选：修改默认参数
     generator.set_default_params(steps=25, width=800, height=1000)
-    
     # 生成图像
-    generator.generate_image(
+    obs_url = generator.generate_image(
         prompt="filmfotos, Asian portrait,A young woman wearing a green baseball cap,covering one eye with her hand",
         reference_image_url="https://dsideal.obs.cn-north-1.myhuaweicloud.com/HuangHai/Backup/HuangWanQiao.jpg",
         control_weight=0.8
-    )
+    )
+    if obs_url:
+        print(f"最终OBS地址: {obs_url}")