qubvel-hf HF staff commited on
Commit
f8c3b5d
·
verified ·
1 Parent(s): 5707a98

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(mobileone_s1, encoder_weights="imagenet")
34
  ```
35
 
36
  ### References
 
30
  ```python
31
  import segmentation_models_pytorch as smp
32
 
33
+ model = smp.Unet("mobileone_s1", encoder_weights="imagenet")
34
  ```
35
 
36
  ### References