Update app.py

app.py

@@ -35,6 +35,7 @@ from transformers import DetrForObjectDetection, DetrImageProcessor
 from pose_render_utils import visualize_keypoints
 
 
+# -----------------------------------------------------------------------------
 # -----------------------------------------------------------------------------
 # Config
 
@@ -42,26 +43,26 @@ ASSETS_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), "assets")
 CONFIGS_DIR = os.path.join(ASSETS_DIR, "configs")
 
 POSE_MODELS = {
-    "0.4B": {
-        "repo": "facebook/sapiens2-pose-0.4b",
-        "filename": "sapiens2_0.4b_pose.safetensors",
-        "config": os.path.join(CONFIGS_DIR, "sapiens2_0.4b_keypoints308_shutterstock_goliath_3po-1024x768.py"),
-    },
-    "0.8B": {
-        "repo": "facebook/sapiens2-pose-0.8b",
-        "filename": "sapiens2_0.8b_pose.safetensors",
-        "config": os.path.join(CONFIGS_DIR, "sapiens2_0.8b_keypoints308_shutterstock_goliath_3po-1024x768.py"),
-    },
+#     "0.4B": {
+#         "repo": "facebook/sapiens2-pose-0.4b",
+#         "filename": "sapiens2_0.4b_pose.safetensors",
+#         "config": os.path.join(CONFIGS_DIR, "sapiens2_0.4b_keypoints308_shutterstock_goliath_3po-1024x768.py"),
+#     },
+#     "0.8B": {
+#         "repo": "facebook/sapiens2-pose-0.8b",
+#         "filename": "sapiens2_0.8b_pose.safetensors",
+#         "config": os.path.join(CONFIGS_DIR, "sapiens2_0.8b_keypoints308_shutterstock_goliath_3po-1024x768.py"),
+#     },
     "1B": {
         "repo": "facebook/sapiens2-pose-1b",
         "filename": "sapiens2_1b_pose.safetensors",
         "config": os.path.join(CONFIGS_DIR, "sapiens2_1b_keypoints308_shutterstock_goliath_3po-1024x768.py"),
     },
-    "5B": {
-        "repo": "facebook/sapiens2-pose-5b",
-        "filename": "sapiens2_5b_pose.safetensors",
-        "config": os.path.join(CONFIGS_DIR, "sapiens2_5b_keypoints308_shutterstock_goliath_3po-1024x768.py"),
-    },
+#     "5B": {
+#         "repo": "facebook/sapiens2-pose-5b",
+#         "filename": "sapiens2_5b_pose.safetensors",
+#         "config": os.path.join(CONFIGS_DIR, "sapiens2_5b_keypoints308_shutterstock_goliath_3po-1024x768.py"),
+#     },
 }
 DEFAULT_SIZE = "1B"