Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
7ccb658
1
Parent(s):
de0b7d0
Upgrade the weights loading method to avoid duplicated loading.
Browse files
app.py
CHANGED
@@ -52,6 +52,7 @@ birefnet.eval()
|
|
52 |
|
53 |
@spaces.GPU
|
54 |
def predict(image, resolution, weights_file):
|
|
|
55 |
if weights_file != 'General':
|
56 |
# Load BiRefNet with chosen weights
|
57 |
_weights_file = '/'.join(('zhengpeng7', usage_to_weights_file[weights_file] if weights_file is not None else 'BiRefNet'))
|
|
|
52 |
|
53 |
@spaces.GPU
|
54 |
def predict(image, resolution, weights_file):
|
55 |
+
global birefnet
|
56 |
if weights_file != 'General':
|
57 |
# Load BiRefNet with chosen weights
|
58 |
_weights_file = '/'.join(('zhengpeng7', usage_to_weights_file[weights_file] if weights_file is not None else 'BiRefNet'))
|