Update README.md

README.md

@@ -5,18 +5,28 @@ pipeline_tag: mask-generation
 
 This model repository contains the weights of the MobileSAM model.
 
+## Installation
+
+First install the M
+
 ## Usage
 
-The model can be used as follows:
+First install the MobileSAM package:
 
-```python
-!pip install git+https://github.com/ChaoningZhang/MobileSAM.git
+```bash
+pip install git+https://github.com/ChaoningZhang/MobileSAM.git
+```
 
+The model can then be used as follows:
+
+```python
 from mobile_sam import MobileSAM, SamPredictor
+import torch
 
 model = MobileSAM.from_pretrained("nielsr/mobilesam")
 
 # perform inference
+device = "cuda" if torch.cuda.is_available() else "cpu"
 model.to(device=device)
 
 predictor = SamPredictor(model)