Update app.py

app.py

@@ -44,7 +44,7 @@ class SegmentationRequest(BaseModel):
     
 image_url="https://res.cloudinary.com/dyvuvklpk/image/upload/v1751009512/MEN-Denim-id_00000089-46_7_additional_ow2h0l.png"
 #print(hi)
-print(image_url)
+
 class SegmentInfo(BaseModel):
     class_id: int
     class_name: str
@@ -116,7 +116,7 @@ def download_image(url: str) -> Image.Image:
             image = image.convert('RGB')
         
         
-        print("Image downloaaded succcessfully:",url)
+        logger.info("Image downloaaded succcessfully:",url)
         
         return image
     except Exception as e:
@@ -146,10 +146,7 @@ def get_category_folder(class_id):
     """Get appropriate folder for class"""
     if class_id in CLOTHING_ITEMS:
         return "clothing"
-    elif class_id in ACCESSORIES:
-        return "accessories"
-    elif class_id in BODY_PARTS:
-        return "body_parts"
+    
     else:
         return "clothing"  # default