Update Videobook/Videobook.py

Videobook/Videobook.py

@@ -42,7 +42,7 @@ class Videobook:
         imgs_buff.append(img)
     return imgs_buff
 
-  def imgs_to_video(self, imgs, video_name='video.mp4'):
+  def imgs_to_video(self, imgs, video_name='video.mp4v'):
     video_dims = (imgs[0].width, imgs[0].height)
     fourcc = cv2.VideoWriter_fourcc(*'DIVX')    
     video = cv2.VideoWriter(video_name, fourcc, self.fps, video_dims)
@@ -51,10 +51,10 @@ class Videobook:
       video.write(cv2.cvtColor(np.array(tmp_img), cv2.COLOR_RGB2BGR))
     video.release()
 
-  def make_video(self, imgs, lengths, video_name = "finished_video.mp4"):
-    self.imgs_to_video(self.addBuffer(imgs, lengths), 'test_video.mp4')
+  def make_video(self, imgs, lengths, video_name = "finished_video.mp4v"):
+    self.imgs_to_video(self.addBuffer(imgs, lengths), 'test_video.mp4v')
     input_audio = ffmpeg.input(os.path.join(os.getcwd(),'tts.mp3'))
-    input_video = ffmpeg.input('/content/test_video.mp4')
+    input_video = ffmpeg.input('/content/test_video.mp4v')
     ffmpeg.concat(input_video, input_audio, v=1, a=1).output(video_name).run(overwrite_output=True)
     
 
@@ -70,4 +70,4 @@ class Videobook:
     lengths = self.generate_voice(story, sentences)
     images = self.generate_imgs(sentences, steps)
     self.make_video(images, lengths)
-    return "finished_video.mp4"
+    return "finished_video.mp4v"