Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
@@ -41,7 +41,7 @@ model = UNet(
|
|
41 |
)
|
42 |
|
43 |
# Загрузка весов
|
44 |
-
model_path = hf_hub_download(repo_id="
|
45 |
model.load_state_dict(torch.load(model_path, map_location='cpu'))
|
46 |
model.eval()
|
47 |
```
|
|
|
41 |
)
|
42 |
|
43 |
# Загрузка весов
|
44 |
+
model_path = hf_hub_download(repo_id="Nikitapyscript/MRISegment_model", filename="best_metric_model.pth")
|
45 |
model.load_state_dict(torch.load(model_path, map_location='cpu'))
|
46 |
model.eval()
|
47 |
```
|