Hot-Fix: 수정 process 함수 정의 및 호출

AI-SIP · Nov 24, 2024 · c282680 · c282680
1 parent 6c07124
commit c282680
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/app/AIProcessor.py b/app/AIProcessor.py
@@ -150,7 +150,7 @@ def combine_alpha(self, image_rgb):
         else:
             return image_rgb
 
-    def process(self, img_bytes, user_inputs, user_labels, extension='jpg'):
+    def process(self, img_bytes, user_inputs, extension='jpg'):
         """ local test 용도 Vs. server test 용도 구분 """
         ### local 용도
         # img_path = img_bytes

diff --git a/app/main.py b/app/main.py
@@ -126,8 +126,7 @@ async def processShape(request: Request):
         # aiProcessor = AIProcessor(yolo_path='/Users/semin/models/yolo11_best.pt', sam_path='/Users/semin/models/mobile_sam.pt')  # local
         aiProcessor = AIProcessor(yolo_path="../models/yolo11_best.pt", sam_path="../models/mobile_sam.pt")  # server
         img_input_bytes, img_mask_bytes, img_output_bytes, one, two = aiProcessor.process(img_bytes=corrected_img_bytes,
-                                                                                          user_points=point_list,
-                                                                                          user_labels=label_list)
+                                                                                          user_inputs=point_list)
         logger.info("AI 필기 제거 프로세스 완료")
 
         upload_image_to_s3(img_input_bytes, paths["input_path"])