Spaces:
Runtime error
Runtime error
Ahsen Khaliq
commited on
Commit
·
bab98df
1
Parent(s):
83f3811
Update app.py
Browse files
app.py
CHANGED
@@ -1,6 +1,7 @@
|
|
1 |
import os
|
2 |
os.system('pip install gradio --upgrade')
|
3 |
os.system('pip freeze')
|
|
|
4 |
import torch
|
5 |
import gradio as gr
|
6 |
from moviepy.editor import *
|
@@ -27,7 +28,7 @@ def inference(video):
|
|
27 |
output_alpha="pha.mp4", # [Optional] Output the raw alpha prediction.
|
28 |
output_foreground="fgr.mp4", # [Optional] Output the raw foreground prediction.
|
29 |
output_video_mbps=4, # Output video mbps. Not needed for png sequence.
|
30 |
-
seq_chunk=
|
31 |
num_workers=1, # Only for image sequence input. Reader threads.
|
32 |
progress=True # Print conversion progress.
|
33 |
)
|
|
|
1 |
import os
|
2 |
os.system('pip install gradio --upgrade')
|
3 |
os.system('pip freeze')
|
4 |
+
os.system('nvidia-smi')
|
5 |
import torch
|
6 |
import gradio as gr
|
7 |
from moviepy.editor import *
|
|
|
28 |
output_alpha="pha.mp4", # [Optional] Output the raw alpha prediction.
|
29 |
output_foreground="fgr.mp4", # [Optional] Output the raw foreground prediction.
|
30 |
output_video_mbps=4, # Output video mbps. Not needed for png sequence.
|
31 |
+
seq_chunk=7, # Process n frames at once for better parallelism.
|
32 |
num_workers=1, # Only for image sequence input. Reader threads.
|
33 |
progress=True # Print conversion progress.
|
34 |
)
|