kevinwang676 commited on
Commit
7590016
1 Parent(s): bc20090

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -125,6 +125,7 @@ def inference_video(video_path, prompt):
125
  # pbar.set_description("generating video: ")
126
  for i, file in enumerate(input_frames):
127
  img = cv2.imread(os.path.join("demo_temp/input_frames", file))
 
128
  #vis = img + np.array([0, 0, 128]) * output[i][1].transpose(1, 2, 0)
129
  vis = img + output[i][1].transpose(1,2,0) * 255
130
  vis = np.clip(vis, 0, 255)
 
125
  # pbar.set_description("generating video: ")
126
  for i, file in enumerate(input_frames):
127
  img = cv2.imread(os.path.join("demo_temp/input_frames", file))
128
+ img[:] = 0
129
  #vis = img + np.array([0, 0, 128]) * output[i][1].transpose(1, 2, 0)
130
  vis = img + output[i][1].transpose(1,2,0) * 255
131
  vis = np.clip(vis, 0, 255)