37 lines
1.4 KiB
Python
37 lines
1.4 KiB
Python
from app.schemas.generate_image import BatchGenerateRelightImageModel, BatchGenerateProductImageModel
|
|
from app.service.generate_batch_image.service_batch_generate_product_image import batch_generate_product
|
|
|
|
from app.service.generate_batch_image.service_batch_generate_relight_image import batch_generate_relight
|
|
|
|
if __name__ == '__main__':
|
|
rd = BatchGenerateProductImageModel(
|
|
tasks_id="test1-89",
|
|
image_strength=0.7,
|
|
prompt=" The best quality, masterpiece, real image.Outwear,high quality clothing details,8K realistic,HDR",
|
|
image_url="aida-results/result_40b1a2fe-e220-11ef-9bfa-0242ac150003.png",
|
|
product_type="single",
|
|
batch_size=2
|
|
)
|
|
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())
|