Refactor vis_animals.py to load model using get_model from the registered model registry, enhancing integration with the model registry.

Author: xiu-cs

animals/demo/vis_animals.py

@@ -42,8 +42,9 @@
     spec.loader.exec_module(module)
     CFM = getattr(module, "Model")
 else:
-    # Load model from installed fmpose package
-    from fmpose3d.models import Model as CFM
+    # Load model from registered model registry
+    from fmpose3d.models import get_model
+    CFM = get_model(args.model_type)
 
 from deeplabcut.pose_estimation_pytorch.apis import superanimal_analyze_images