Ash2505 commited on
Commit
d3e5ae0
·
verified ·
1 Parent(s): 58e8622

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -18,6 +18,7 @@ birefnet.half()
18
 
19
  def extract_object(image, t1, t2):
20
  # Data settings
 
21
  image_size = (1024, 1024)
22
  transform_image = transforms.Compose([
23
  transforms.Resize(image_size),
 
18
 
19
  def extract_object(image, t1, t2):
20
  # Data settings
21
+ imageResized = image.resize((512, 512))
22
  image_size = (1024, 1024)
23
  transform_image = transforms.Compose([
24
  transforms.Resize(image_size),