a9 commited on
Commit
56cd167
·
verified ·
1 Parent(s): 25d92a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -47,7 +47,7 @@ NORMAL_MODELS = {
47
  DEFAULT_SIZE = "1B"
48
 
49
  # v1 binary fg/bg TorchScript model.
50
- FG_REPO = "facebook/sapiens-seg-foreground-1b-torchscript"
51
  FG_FILENAME = "sapiens_0.8b_seg_foreground_epoch_8_torchscript.pt2"
52
  BG_OPTIONS = ["fg-bg", "no-bg-removal"]
53
  DEFAULT_BG = "fg-bg"
 
47
  DEFAULT_SIZE = "1B"
48
 
49
  # v1 binary fg/bg TorchScript model.
50
+ FG_REPO = "facebook/sapiens-seg-foreground-0.8b-torchscript"
51
  FG_FILENAME = "sapiens_0.8b_seg_foreground_epoch_8_torchscript.pt2"
52
  BG_OPTIONS = ["fg-bg", "no-bg-removal"]
53
  DEFAULT_BG = "fg-bg"