Spaces:
Running
on
L40S
Running
on
L40S
Update app.py
Browse files
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[
|
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{
|