Spaces:
Runtime error
Runtime error
Update Videobook/Videobook.py
#1
by
Icar
- opened
- Videobook/Videobook.py +5 -5
Videobook/Videobook.py
CHANGED
@@ -42,7 +42,7 @@ class Videobook:
|
|
42 |
imgs_buff.append(img)
|
43 |
return imgs_buff
|
44 |
|
45 |
-
def imgs_to_video(self, imgs, video_name='video.
|
46 |
video_dims = (imgs[0].width, imgs[0].height)
|
47 |
fourcc = cv2.VideoWriter_fourcc(*'DIVX')
|
48 |
video = cv2.VideoWriter(video_name, fourcc, self.fps, video_dims)
|
@@ -51,10 +51,10 @@ class Videobook:
|
|
51 |
video.write(cv2.cvtColor(np.array(tmp_img), cv2.COLOR_RGB2BGR))
|
52 |
video.release()
|
53 |
|
54 |
-
def make_video(self, imgs, lengths, video_name = "finished_video.
|
55 |
-
self.imgs_to_video(self.addBuffer(imgs, lengths), 'test_video.
|
56 |
input_audio = ffmpeg.input(os.path.join(os.getcwd(),'tts.mp3'))
|
57 |
-
input_video = ffmpeg.input('/content/test_video.
|
58 |
ffmpeg.concat(input_video, input_audio, v=1, a=1).output(video_name).run(overwrite_output=True)
|
59 |
|
60 |
|
@@ -70,4 +70,4 @@ class Videobook:
|
|
70 |
lengths = self.generate_voice(story, sentences)
|
71 |
images = self.generate_imgs(sentences, steps)
|
72 |
self.make_video(images, lengths)
|
73 |
-
return "finished_video.
|
|
|
42 |
imgs_buff.append(img)
|
43 |
return imgs_buff
|
44 |
|
45 |
+
def imgs_to_video(self, imgs, video_name='video.mp4v'):
|
46 |
video_dims = (imgs[0].width, imgs[0].height)
|
47 |
fourcc = cv2.VideoWriter_fourcc(*'DIVX')
|
48 |
video = cv2.VideoWriter(video_name, fourcc, self.fps, video_dims)
|
|
|
51 |
video.write(cv2.cvtColor(np.array(tmp_img), cv2.COLOR_RGB2BGR))
|
52 |
video.release()
|
53 |
|
54 |
+
def make_video(self, imgs, lengths, video_name = "finished_video.mp4v"):
|
55 |
+
self.imgs_to_video(self.addBuffer(imgs, lengths), 'test_video.mp4v')
|
56 |
input_audio = ffmpeg.input(os.path.join(os.getcwd(),'tts.mp3'))
|
57 |
+
input_video = ffmpeg.input('/content/test_video.mp4v')
|
58 |
ffmpeg.concat(input_video, input_audio, v=1, a=1).output(video_name).run(overwrite_output=True)
|
59 |
|
60 |
|
|
|
70 |
lengths = self.generate_voice(story, sentences)
|
71 |
images = self.generate_imgs(sentences, steps)
|
72 |
self.make_video(images, lengths)
|
73 |
+
return "finished_video.mp4v"
|