Update app.py
Browse files
app.py
CHANGED
@@ -117,7 +117,7 @@ class CloseNet(nn.Module):
|
|
117 |
return self.segm_dec(features)
|
118 |
|
119 |
# Load Pretrained Model
|
120 |
-
model_path = "/
|
121 |
model = CloseNet()
|
122 |
model.load_state_dict(torch.load(model_path, map_location=torch.device("cpu")), strict=False)
|
123 |
model.eval()
|
|
|
117 |
return self.segm_dec(features)
|
118 |
|
119 |
# Load Pretrained Model
|
120 |
+
model_path = "model_arch/closenet.pth"
|
121 |
model = CloseNet()
|
122 |
model.load_state_dict(torch.load(model_path, map_location=torch.device("cpu")), strict=False)
|
123 |
model.eval()
|