Files
AiDA_Python/app/service/generate_batch_image/test.py

32 lines
1.1 KiB
Python
Raw Normal View History

from app.schemas.generate_image import BatchGenerateProductImageModel
from app.service.generate_batch_image.service_batch_generate_product_image import batch_generate_product
if __name__ == '__main__':
rd = BatchGenerateProductImageModel(
batch_tasks_id="",
batch_data_list="",
user_id=""
)
x = batch_generate_product.delay(rd.dict())
print(x)
"""relight"""
# rd = BatchGenerateRelightImageModel(
# tasks_id="123-89",
# # prompt="beautiful woman, detailed face, sunshine, outdoor, warm atmosphere",
# prompt="Colorful black",
# image_url='aida-results/result_0000b606-1902-11ef-9424-0242ac180002.png',
# direction="Right Light",
# product_type="single",
# batch_size=2
# )
# batch_generate_relight.delay(rd.dict())
"""pose transform"""
# rd = BatchPoseTransformModel(
# tasks_id="123-89",
# image_url='aida-results/result_0000b606-1902-11ef-9424-0242ac180002.png',
# pose_id="1",
# batch_size=10
# )
# batch_pose_transform_tasks.delay(rd.dict())