Update BiRefNet.py

BiRefNet.py

@@ -7,10 +7,10 @@ from huggingface_hub import PyTorchModelHubMixin
 
 from config import Config
 from dataset import class_labels_TR_sorted
-from models.backbones.build_backbone import build_backbone
-from models.modules.decoder_blocks import BasicDecBlk, ResBlk
-from models.modules.lateral_blocks import BasicLatBlk
-from models.modules.aspp import ASPP, ASPPDeformable
+from build_backbone import build_backbone
+from decoder_blocks import BasicDecBlk, ResBlk
+from lateral_blocks import BasicLatBlk
+from aspp import ASPP, ASPPDeformable
 
 
 def image2patches(image, grid_h=2, grid_w=2, patch_ref=None, transformation='b c (hg h) (wg w) -> (b hg wg) c h w'):