Update app.py

app.py

@@ -42,23 +42,13 @@ NORMAL_MODELS = {
         "repo": "facebook/sapiens2-normal-0.8b",
         "filename": "sapiens2_0.8b_normal.safetensors",
         "config": os.path.join(CONFIGS_DIR, "sapiens2_0.8b_normal_metasim_render_people-1024x768.py"),
-    },
-    "1B": {
-        "repo": "facebook/sapiens2-normal-1b",
-        "filename": "sapiens2_1b_normal.safetensors",
-        "config": os.path.join(CONFIGS_DIR, "sapiens2_1b_normal_metasim_render_people-1024x768.py"),
-    },
-    "5B": {
-        "repo": "facebook/sapiens2-normal-5b",
-        "filename": "sapiens2_5b_normal.safetensors",
-        "config": os.path.join(CONFIGS_DIR, "sapiens2_5b_normal_metasim_render_people-1024x768.py"),
-    },
+    }
 }
 DEFAULT_SIZE = "1B"
 
 # v1 binary fg/bg TorchScript model.
 FG_REPO = "facebook/sapiens-seg-foreground-1b-torchscript"
-FG_FILENAME = "sapiens_1b_seg_foreground_epoch_8_torchscript.pt2"
+FG_FILENAME = "sapiens_0.8b_seg_foreground_epoch_8_torchscript.pt2"
 BG_OPTIONS = ["fg-bg", "no-bg-removal"]
 DEFAULT_BG = "fg-bg"