Spaces:

SerdarHelli
/

deprem_satellite_semantic_opencitieschallange

Sleeping

SerdarHelli commited on Feb 11, 2023

Commit

1ecf700

1 Parent(s): e164917

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -95,7 +95,7 @@ model = smp.Unet(
 device='cuda' if torch.cuda.is_available() else 'cpu'
 map_location=torch.device(device)
-state_dict = torch.load("/content/models/stage3/srx50-f0/checkpoints/best.pth",map_location=map_location)["state_dict"]

 device='cuda' if torch.cuda.is_available() else 'cpu'
 map_location=torch.device(device)
+state_dict = torch.load("srx50-f0.pth",map_location=map_location)["state_dict"]