qubvel-hf HF staff commited on
Commit
dc2020e
·
verified ·
1 Parent(s): b9520cc

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -30,7 +30,7 @@ pip install segmentation-models-pytorch
30
  ```python
31
  import segmentation_models_pytorch as smp
32
 
33
- model = smp.Unet(resnet50, encoder_weights="swsl")
34
  ```
35
 
36
  ### References
 
30
  ```python
31
  import segmentation_models_pytorch as smp
32
 
33
+ model = smp.Unet("resnet50", encoder_weights="swsl")
34
  ```
35
 
36
  ### References