Commit
·
55219fd
1
Parent(s):
c21af4b
yolo fix
Browse files
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", "
|
| 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"]:
|