Spaces:

gaur3009
/

detecting_dress

Running

gaur3009 commited on Feb 25

Commit

ac99583

verified ·

1 Parent(s): 32a5206

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ import torch
 import cv2
 from PIL import Image
 from torchvision import transforms
-from cloth-segmentation.networks.u2net import U2NET  # Import U²-Net
 # Load U²-Net model
 model_path = "cloth-segmentation/models/u2net.pth"  # Ensure this path is correct

 import cv2
 from PIL import Image
 from torchvision import transforms
+from cloth_segmentation.networks.u2net import U2NET  # Import U²-Net
 # Load U²-Net model
 model_path = "cloth-segmentation/models/u2net.pth"  # Ensure this path is correct