Spaces:
Running
Running
Update app.py
Browse files
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 |
|