diff --git a/app/service/generate_image/service.py b/app/service/generate_image/service.py
index 0292223..081d361 100644
--- a/app/service/generate_image/service.py
+++ b/app/service/generate_image/service.py
@@ -85,7 +85,7 @@ class GenerateImage:
                 # 污点检测
                 is_smudge, not_smudge_image = stain_detection(remove_bg_image)
                 # 类型识别
-                category, scores, not_smudge_image = generate_category_recognition(image=image_result, gender=self.gender)
+                category, scores, not_smudge_image = generate_category_recognition(image=remove_bg_image, gender=self.gender)
                 self.generate_data['category'] = str(category)
                 image_result = not_smudge_image
             if is_smudge:  # 无污点