Update app.py
Browse files
app.py
CHANGED
@@ -10,7 +10,8 @@ from PIL import Image
|
|
10 |
from typing import Tuple
|
11 |
|
12 |
net=BriaRMBG()
|
13 |
-
model_path = "./model1.pth"
|
|
|
14 |
if torch.cuda.is_available():
|
15 |
net.load_state_dict(torch.load(model_path))
|
16 |
net=net.cuda()
|
|
|
10 |
from typing import Tuple
|
11 |
|
12 |
net=BriaRMBG()
|
13 |
+
# model_path = "./model1.pth"
|
14 |
+
model_path = hf_hub_download("briaai/RMBG-1.4", 'model.pth')
|
15 |
if torch.cuda.is_available():
|
16 |
net.load_state_dict(torch.load(model_path))
|
17 |
net=net.cuda()
|