updated README. fix tests.

Author: n-poulsen

README.md

@@ -51,6 +51,8 @@ listed using the `dlclibrary.get_available_detectors` and
 
 ## How to add a new model?
 
+### TensorFlow models
+
 Pick a good model_name. Follow the (novel) naming convention (modeltype_species), e.g. ```superanimal_topviewmouse```.  
 
 1. Add the model_name with path and commit ID to: https://github.com/DeepLabCut/DLClibrary/blob/main/dlclibrary/dlcmodelzoo/modelzoo_urls.yaml
@@ -59,3 +61,30 @@ Pick a good model_name. Follow the (novel) naming convention (modeltype_species)
 https://github.com/DeepLabCut/DLClibrary/blob/main/dlclibrary/dlcmodelzoo/modelzoo_download.py#L15
 
 3. For superanimal models also fill in the configs!
+
+### PyTorch models (for `deeplabcut >= 3.0.0`)
+
+PyTorch models are listed in [`dlclibrary/dlcmodelzoo/modelzoo_urls_pytorch.yaml`](
+https://github.com/DeepLabCut/DLClibrary/blob/main/dlclibrary/dlcmodelzoo/modelzoo_urls_pytorch.yaml
+). The file is organized as:
+
+```yaml
+my_cool_dataset:  # name of the dataset used to train the model
+  detectors:
+    detector_name: path/to/huggingface-detector.pt  # add detectors under `detector`
+  pose_models:
+    pose_model_name: path/to/huggingface-pose-model.pt  # add pose models under `pose_models`
+    other_pose_model_name: path/to/huggingface-other-pose-model.pt
+```
+
+This will allow users to download the models using the format `datatsetName_modelName`,
+i.e. for this example 3 models would be available: `my_cool_dataset_detector_name`,
+`my_cool_dataset_pose_model_name` and `my_cool_dataset_other_pose_model_name`.
+
+To add a new model for `deeplabcut >= 3.0.0`, simply:
+
+- add a new line under detectors or pose models if the dataset is already defined
+- add the structure if the model was trained on a new dataset 
+
+The models will then be listed when calling `dlclibrary.get_available_detectors` or
+`dlclibrary.get_available_models`!

dlclibrary/dlcmodelzoo/modelzoo_download.py

@@ -28,7 +28,6 @@
     "mouse_pupil_vclose",
     "horse_sideview",
     "full_macaque",
-    "superanimal_bird",
     "superanimal_quadruped",
     "superanimal_topviewmouse",
 ]