File size: 383 Bytes
9e754a4
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
from API import aud2text, Summarize
from moviepy.editor import *

class Vid2SumModel:

    def predict(self, vid_file):
        video = VideoFileClip(vid_file)
        audio_file = "converted_audio.mp3"
        video.audio.write_audiofile(audio_file)

        text = self.aud2text(audio_file)
        sum_text = self.summarize(text["text"])
        return sum_text[0]["summary_text"]