vibs08 commited on
Commit
7e664ff
·
verified ·
1 Parent(s): 53b232b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -164,8 +164,8 @@ def preprocess(input_image, do_remove_background, foreground_ratio):
164
  image = resize_foreground(image, foreground_ratio)
165
  image = fill_background(image)
166
 
167
- torch.cuda.synchronize()
168
- torch.cuda.empty_cache()
169
  print("Background Removed")
170
  else:
171
  image = input_image
 
164
  image = resize_foreground(image, foreground_ratio)
165
  image = fill_background(image)
166
 
167
+ # torch.cuda.synchronize()
168
+ # torch.cuda.empty_cache()
169
  print("Background Removed")
170
  else:
171
  image = input_image