Update README.md
Browse files
README.md
CHANGED
|
@@ -50,7 +50,7 @@ import torch as t
|
|
| 50 |
from huggingface_hub import hf_hub_download
|
| 51 |
|
| 52 |
# 1. Download the architecture and weights files
|
| 53 |
-
REPO_ID = "
|
| 54 |
hf_hub_download(repo_id=REPO_ID, filename="model.py", local_dir=".")
|
| 55 |
hf_hub_download(repo_id=REPO_ID, filename="model_weights_hybrid.pth", local_dir=".")
|
| 56 |
|
|
|
|
| 50 |
from huggingface_hub import hf_hub_download
|
| 51 |
|
| 52 |
# 1. Download the architecture and weights files
|
| 53 |
+
REPO_ID = "ALEXFLR/sbl-net"
|
| 54 |
hf_hub_download(repo_id=REPO_ID, filename="model.py", local_dir=".")
|
| 55 |
hf_hub_download(repo_id=REPO_ID, filename="model_weights_hybrid.pth", local_dir=".")
|
| 56 |
|