Update app.py

app.py

@@ -42,13 +42,18 @@ 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"),
+    },
 }
 DEFAULT_SIZE = "1B"
 
 # v1 binary fg/bg TorchScript model.
-FG_REPO = "facebook/sapiens-seg-foreground-0.8b-torchscript"
-FG_FILENAME = "sapiens_0.8b_seg_foreground_epoch_8_torchscript.pt2"
+FG_REPO = "facebook/sapiens-seg-foreground-1b-torchscript"
+FG_FILENAME = "sapiens_1b_seg_foreground_epoch_8_torchscript.pt2"
 BG_OPTIONS = ["fg-bg", "no-bg-removal"]
 DEFAULT_BG = "fg-bg"