Update README.md
Browse files
README.md
CHANGED
|
@@ -6,5 +6,5 @@ To load the pretrained model:
|
|
| 6 |
```
|
| 7 |
from exlib.datasets.massmaps import MassMapsConvnetRegModel
|
| 8 |
|
| 9 |
-
model = MassMapsConvnetForImageRegression.from_pretrained(f'BrachioLab/
|
| 10 |
```
|
|
|
|
| 6 |
```
|
| 7 |
from exlib.datasets.massmaps import MassMapsConvnetRegModel
|
| 8 |
|
| 9 |
+
model = MassMapsConvnetForImageRegression.from_pretrained(f'BrachioLab/massmaps-conv')
|
| 10 |
```
|