import torch
from diffusers import StableDiffusionPipeline
import mediapipe as mp

class EliteMotionModel:
    def __init__(self):
        self.model = StableDiffusionPipeline.from_pretrained("BootyShakerAI")
        self.motion_tracker = mp.solutions.pose.Pose(
            min_detection_confidence=0.8,
            min_tracking_confidence=0.8
        )
        self.device = "cuda" if torch.cuda.is_available() else "cpu"
        self.model.to(self.device)