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 |
```
|