Spaces:

aryanxxvii
/

clearbg_space

Sleeping

Aryan Wadhawan commited on 28 days ago

Commit

973996a

•

1 Parent(s): 9aa8e63

Add model

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ from torchvision import transforms
 from PIL import Image
 import numpy as np
 import os
-from clearbg_space.u2net import U2NET
 import data_transforms
 import torch.nn.functional as F
 from skimage import io

 from PIL import Image
 import numpy as np
 import os
+from u2net import U2NET
 import data_transforms
 import torch.nn.functional as F
 from skimage import io