| 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) | |