Spaces:
Runtime error
Runtime error
kevinwang676
commited on
Commit
•
7590016
1
Parent(s):
bc20090
Update app.py
Browse files
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)
|