Spaces:

build-small-hackathon
/

Pozify

Paused

Pozify / src /pozify /steps /pose_backends /base.py

fix(hf-spaces): install mediapipe native libraries

6d4caef 5 months ago

614 Bytes

from __future__ import annotations

from dataclasses import dataclass
from typing import Any, Protocol


@dataclass(frozen=True)
class PoseDetection:
    landmarks: dict[str, dict[str, float]]
    world_landmarks: dict[str, dict[str, float]]
    source: str


class PoseBackendUnavailableError(RuntimeError):
    pass


class PoseBackend(Protocol):
    source: str

    def __enter__(self) -> "PoseBackend":
        return self

    def __exit__(self, *args: object) -> None:
        return None

    def detect(self, rgb_frame: Any | None, *, frame_index: int) -> PoseDetection:
        raise NotImplementedError

	from __future__ import annotations

	from dataclasses import dataclass
	from typing import Any, Protocol


	@dataclass(frozen=True)
	class PoseDetection:
	landmarks: dict[str, dict[str, float]]
	world_landmarks: dict[str, dict[str, float]]
	source: str


	class PoseBackendUnavailableError(RuntimeError):
	pass


	class PoseBackend(Protocol):
	source: str

	def __enter__(self) -> "PoseBackend":
	return self

	def __exit__(self, *args: object) -> None:
	return None

	def detect(self, rgb_frame: Any \| None, *, frame_index: int) -> PoseDetection:
	raise NotImplementedError