ZhengPeng7 commited on
Commit
fbe03e2
·
1 Parent(s): a10635a

Add weights option to BiRefNet trained in all different settings.

Browse files
app.py CHANGED
@@ -50,8 +50,7 @@ usage_to_weights_file = {
50
  'DIS-TR_TEs': 'BiRefNet-DIS5K-TR_TEs'
51
  }
52
 
53
- # def predict(image_1, image_2):
54
- # images = [image_1, image_2]
55
  @spaces.GPU
56
  def predict(image, resolution, weights_file):
57
  global weights_path
@@ -92,7 +91,7 @@ def predict(image, resolution, weights_file):
92
  return image, image_preds[0]
93
 
94
 
95
- examples = [[_] for _ in glob('materials/examples/*')][:]
96
 
97
  # Add the option of resolution in a text box.
98
  for idx_example, example in enumerate(examples):
 
50
  'DIS-TR_TEs': 'BiRefNet-DIS5K-TR_TEs'
51
  }
52
 
53
+
 
54
  @spaces.GPU
55
  def predict(image, resolution, weights_file):
56
  global weights_path
 
91
  return image, image_preds[0]
92
 
93
 
94
+ examples = [[_] for _ in glob('examples/*')][:]
95
 
96
  # Add the option of resolution in a text box.
97
  for idx_example, example in enumerate(examples):
{materials/examples → examples}/Helicopter.jpg RENAMED
File without changes
{materials/examples → examples}/Jewelry.jpg RENAMED
File without changes
{materials/examples → examples}/Windmill.jpg RENAMED
File without changes