| 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"] |