BladeSzaSza commited on
Commit
55219fd
·
1 Parent(s): c21af4b
backend/gradio_labanmovementanalysis/pose_estimation.py CHANGED
@@ -411,7 +411,7 @@ def get_pose_estimator(model_spec: str) -> PoseEstimator:
411
  size = "n" # default size
412
 
413
  if len(parts) > 1: # "yolo-v8" or "yolo-v11"
414
- if parts[1] in ["v8", "v11"]: # Add other versions as needed
415
  version = parts[1]
416
  # If parts[1] is a size (e.g. "yolo-n"), then version remains default "v8" and size is parts[1]
417
  elif parts[1] in ["n", "s", "m", "l", "x"]:
 
411
  size = "n" # default size
412
 
413
  if len(parts) > 1: # "yolo-v8" or "yolo-v11"
414
+ if parts[1] in ["v8", "11"]: # Add other versions as needed
415
  version = parts[1]
416
  # If parts[1] is a size (e.g. "yolo-n"), then version remains default "v8" and size is parts[1]
417
  elif parts[1] in ["n", "s", "m", "l", "x"]: