BRIA-RMBG-1.4

Sleeping

nielsr HF staff commited on Mar 4

Commit

018621a

•

1 Parent(s): b98efed

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,9 +9,9 @@ import PIL
 from PIL import Image
 from typing import Tuple
-net = BriaRMBG.from_pretrained("briaai/RMBG-1.4")
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 net.to(device)

 from PIL import Image
 from typing import Tuple
+net = BriaRMBG()
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+net = BriaRMBG.from_pretrained("briaai/RMBG-1.4")
 net.to(device)