Create Core.py

Core.py

@@ -0,0 +1,13 @@
+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)