diff --git a/app/service/generate_image/service.py b/app/service/generate_image/service.py index 701b1e8..4ae34fd 100644 --- a/app/service/generate_image/service.py +++ b/app/service/generate_image/service.py @@ -75,7 +75,6 @@ class GenerateImage: self.generate_data['data'] = str(error) self.redis_client.set(self.tasks_id, json.dumps(self.generate_data)) else: - logger.info(1111111111111111111111111111) image_result = result.as_numpy("generated_image")[0] is_smudge = True if self.category == "sketch": @@ -86,7 +85,7 @@ class GenerateImage: image_result = not_smudge_image if is_smudge: # 无污点 image_url = upload_png_sd(image_result, user_id=self.user_id, category=f"{self.category}", object_name=f"{self.tasks_id}.png") - logger.info(f"upload image SUCCESS : {image_url}") + # logger.info(f"upload image SUCCESS : {image_url}") self.generate_data['status'] = "SUCCESS" self.generate_data['message'] = "success" self.generate_data['data'] = str(image_url) @@ -96,7 +95,7 @@ class GenerateImage: self.generate_data['message'] = "success" self.generate_data['data'] = str(GI_SYS_IMAGE_URL) self.redis_client.set(self.tasks_id, json.dumps(self.generate_data)) - logger.info(f"stain_detection result : {self.generate_data}") + # logger.info(f"stain_detection result : {self.generate_data}") def read_tasks_status(self): status_data = self.redis_client.get(self.tasks_id) @@ -133,7 +132,7 @@ class GenerateImage: generate_data = None while time_out > 0: generate_data, _ = self.read_tasks_status() - logger.info(generate_data) + # logger.info(generate_data) if generate_data['status'] in ["REVOKED", "FAILURE"]: ctx.cancel() break @@ -141,7 +140,7 @@ class GenerateImage: break time_out -= 1 time.sleep(0.1) - logger.info(time_out, generate_data) + # logger.info(time_out, generate_data) return generate_data except Exception as e: self.generate_data['status'] = "FAILURE"