Kaushik066 commited on
Commit
4e39487
·
1 Parent(s): 42e6100

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -247,7 +247,7 @@ def tensor_to_video(video_tensor, fps=30, output_path="output.mp4"):
247
  fourcc = cv2.VideoWriter_fourcc(*"mp4v") # MP4 Codec
248
  out = cv2.VideoWriter(output_path, fourcc, fps, (width, height))
249
 
250
- for frame in prod_video:
251
  out.write(frame)
252
  out.release()
253
 
 
247
  fourcc = cv2.VideoWriter_fourcc(*"mp4v") # MP4 Codec
248
  out = cv2.VideoWriter(output_path, fourcc, fps, (width, height))
249
 
250
+ for frame in video_numpy:
251
  out.write(frame)
252
  out.release()
253