Ahsen Khaliq commited on
Commit
bab98df
·
1 Parent(s): 83f3811

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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=1, # Process n frames at once for better parallelism.
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
  )