Hugging Face's logo

Spaces:
briaai
/
BRIA-RMBG-1.4
Running

App Files Community
4
nielsr HF Staff commited on
Commit
018621a
·
verified ·
1 Parent(s): b98efed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,9 +9,9 @@ import PIL
9
  from PIL import Image
10
  from typing import Tuple
11
 
12
- net = BriaRMBG.from_pretrained("briaai/RMBG-1.4")
13
-
14
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
15
  net.to(device)
16
 
17
 
 
9
  from PIL import Image
10
  from typing import Tuple
11
 
12
+ net = BriaRMBG()
 
13
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
14
+ net = BriaRMBG.from_pretrained("briaai/RMBG-1.4")
15
  net.to(device)
16
 
17