Spaces:

RihemXX
/

wrinkle-segmentation-unet

Running

Rihem02 commited on Apr 18

Commit

54ba009

1 Parent(s): f64a081

fixed

Files changed (2) hide show

app.py CHANGED Viewed

@@ -20,7 +20,7 @@ device = torch.device("cpu")
 print("Using device:", device)
 # Load weights into UNet model with ResNet34 backbone
-model_path = "best_unet_model_complete.pth"
 if os.path.exists(model_path):
     model = smp.Unet(
         encoder_name="resnet34",

 print("Using device:", device)
 # Load weights into UNet model with ResNet34 backbone
+model_path = "best_unet.pth"
 if os.path.exists(model_path):
     model = smp.Unet(
         encoder_name="resnet34",

best_unet_model_complete.pth → best_unet.pth RENAMED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:47d03022dfb75ed544b70b3b67ce863bf476b2e97201a36ad3c1515c89c7663b
-size 97950016

 version https://git-lfs.github.com/spec/v1
+oid sha256:380e417f7e5dd35d4e2e372950c51a01299fc2e3a7aa8f4db56f60fee9ce0862
+size 97921128