feat generate product image 部署

app/service/generate_image/service_generate_single_logo.py

@@ -21,7 +21,7 @@ from tritonclient.utils import np_to_triton_dtype
 from app.core.config import *
 import tritonclient.grpc as grpcclient
 from app.schemas.generate_image import GenerateSingleLogoImageModel
-from app.service.generate_image.utils.upload_sd_image import upload_png_sd, upload_single_logo
+from app.service.generate_image.utils.upload_sd_image import upload_png_sd, upload_SDXL_image
 
 logger = logging.getLogger()
 
@@ -67,7 +67,7 @@ class GenerateSingleLogoImage:
         else:
             image = result.as_numpy("generated_image")
             image_result = Image.fromarray(np.squeeze(image.astype(np.uint8)))
-            image_url = upload_single_logo(image_result, user_id=self.user_id, category=f"{self.category}", object_name=f"{self.tasks_id}.png")
+            image_url = upload_SDXL_image(image_result, user_id=self.user_id, category=f"{self.category}", object_name=f"{self.tasks_id}.png")
             self.gen_single_logo_data['status'] = "SUCCESS"
             self.gen_single_logo_data['message'] = "success"
             self.gen_single_logo_data['image_url'] = str(image_url)
@@ -131,7 +131,7 @@ if __name__ == '__main__':
     rd = GenerateSingleLogoImageModel(
         tasks_id="123-89",
         prompt='an apple',
-        seed="1",
+        seed="2",
     )
     server = GenerateSingleLogoImage(rd)
     print(server.get_result())