add function "adjust_contrast" in the function callback for better image displaying.

app/service/generate_image/service.py

@@ -23,7 +23,7 @@ from tritonclient.utils import np_to_triton_dtype
 from app.core.config import *
 from app.schemas.generate_image import GenerateImageModel
 from app.service.generate_image.utils.upload_sd_image import upload_png_sd
-
+from app.service.generate_image.utils.adjust_contrast import adjust_contrast
 logger = logging.getLogger()
 
 
@@ -71,6 +71,7 @@ class GenerateImage:
             self.redis_client.set(self.tasks_id, json.dumps(self.generate_data))
         else:
             image_result = result.as_numpy("generated_image")[0]
+            image_result = adjust_contrast(image_result)
             image_url = upload_png_sd(image_result, user_id=self.user_id, category=f"{self.category}", object_name=f"{self.tasks_id}.png")
             self.generate_data['status'] = "SUCCESS"
             self.generate_data['message'] = "success"