Updated for new model label maps

app.py

@@ -12,17 +12,22 @@ app = FastAPI()
 device = torch.device('cpu')  # Hugging Face Space with no GPU
 
 workout_label_map = {
-    0: "Cardio",
-    1: "Strength",
-    2: "Yoga",
-    3: "HIIT"
+    0: "Chest",
+    1: "Back",
+    2: "Legs",
+    3: "Shoulders",
+    4: "Arms",
+    5: "Core",
+    6: "Full Body",
+    7: "Cardio"
 }
 
 mood_label_map = {
     0: "Energized",
     1: "Tired",
     2: "Stressed",
-    3: "Motivated"
+    3: "Motivated",
+    4: "Neutral"
 }
 
 soreness_label_map = {