yasvanthkumar commited on
Commit
88bd56c
·
verified ·
1 Parent(s): 98d2056

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -9
app.py CHANGED
@@ -1,5 +1,5 @@
 
1
  import os
2
- import re
3
  import tempfile
4
  import streamlit as st
5
  import cv2
@@ -8,7 +8,6 @@ import whisper
8
  from moviepy.editor import *
9
  from moviepy.video.fx.all import crop
10
  from scenedetect import detect, ContentDetector
11
- from pytube import YouTube
12
  import warnings
13
 
14
  warnings.filterwarnings("ignore")
@@ -18,13 +17,6 @@ os.environ["TOKENIZERS_PARALLELISM"] = "false"
18
  def load_whisper_model():
19
  return whisper.load_model("base")
20
 
21
- def download_youtube_video(url):
22
- yt = YouTube(url)
23
- stream = yt.streams.filter(progressive=True, file_extension='mp4').order_by('resolution').desc().first()
24
- temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
25
- stream.download(filename=temp_file.name)
26
- return temp_file.name
27
-
28
  def detect_engaging_scenes(video_path, num_scenes=5):
29
  scene_list = detect(video_path, ContentDetector(min_scene_len=15))
30
  return [(s[0].get_seconds(), s[1].get_seconds()) for s in scene_list][:num_scenes]
@@ -189,5 +181,7 @@ def main():
189
 
190
  if __name__ == "__main__":
191
  main()
 
 
192
 
193
 
 
1
+ # 👇 COPY FROM HERE
2
  import os
 
3
  import tempfile
4
  import streamlit as st
5
  import cv2
 
8
  from moviepy.editor import *
9
  from moviepy.video.fx.all import crop
10
  from scenedetect import detect, ContentDetector
 
11
  import warnings
12
 
13
  warnings.filterwarnings("ignore")
 
17
  def load_whisper_model():
18
  return whisper.load_model("base")
19
 
 
 
 
 
 
 
 
20
  def detect_engaging_scenes(video_path, num_scenes=5):
21
  scene_list = detect(video_path, ContentDetector(min_scene_len=15))
22
  return [(s[0].get_seconds(), s[1].get_seconds()) for s in scene_list][:num_scenes]
 
181
 
182
  if __name__ == "__main__":
183
  main()
184
+ # 👆 TO HERE
185
+
186
 
187