OriLib commited on
Commit
8f942dd
1 Parent(s): f128918

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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()