Update README.md

README.md

@@ -59,6 +59,7 @@ out = model.post_process(prob)
 
 ## 3.1 Run inference for image batch
 ```python
+import torch
 from cellseg_models_pytorch.utils import FileHandler
 
 model.set_inference_mode()
@@ -123,3 +124,16 @@ Cellpose original paper:
   pages = {100–106}
 }
 ```
+
+## Licence
+These model weights are released under the Apache License, Version 2.0 (the "License"). You may obtain a copy of the License at:
+
+http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License.
+
+## Additional Terms
+
+While the Apache 2.0 License grants broad permissions, we kindly request that users adhere to the following guidelines:
+Medical or Clinical Use: This model is not intended for use in medical diagnosis, treatment, or prevention of disease of real patients. It should not be used as a substitute for professional medical advice.
+