Spaces:
Running
on
Zero
Running
on
Zero
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{
|