EdenSw commited on
Commit
937b31b
·
1 Parent(s): 3d11425

Update summarize.py

Browse files
Files changed (1) hide show
  1. summarize.py +7 -9
summarize.py CHANGED
@@ -1,13 +1,11 @@
1
  from API import aud2text, Summarize
2
  from moviepy.editor import *
3
 
4
- class Vid2SumModel:
 
 
 
5
 
6
- def predict(self, vid_file):
7
- video = VideoFileClip(vid_file)
8
- audio_file = "converted_audio.mp3"
9
- video.audio.write_audiofile(audio_file)
10
-
11
- text = self.aud2text(audio_file)
12
- sum_text = self.summarize(text["text"])
13
- return sum_text[0]["summary_text"]
 
1
  from API import aud2text, Summarize
2
  from moviepy.editor import *
3
 
4
+ def Vid2Sum(vid_file):
5
+ video = VideoFileClip(vid_file)
6
+ audio_file = "converted_audio.mp3"
7
+ video.audio.write_audiofile(audio_file)
8
 
9
+ text = aud2text(audio_file)
10
+ sum_text=Summarize(text["text"])
11
+ return sum_text[0]["summary_text"]