fffiloni commited on
Commit
3a09fb3
1 Parent(s): ac6a0bc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -155,7 +155,7 @@ def process_image(input_pil, remove_bg):
155
  shutil.rmtree(temp_dir) # Cleanup temporary folder
156
  print(output_video)
157
  torch.cuda.empty_cache()
158
- return output_video[0], output_objects[1], output_objects[0]
159
 
160
  css="""
161
  div#col-container{
 
155
  shutil.rmtree(temp_dir) # Cleanup temporary folder
156
  print(output_video)
157
  torch.cuda.empty_cache()
158
+ return output_video[0], output_objects[0], output_objects[1]
159
 
160
  css="""
161
  div#col-container{