Caching

README.md

@@ -1,8 +1,8 @@
 ---
 title: Videogenic
 emoji: ✨
-colorFrom: purple
-colorTo: pink
+colorFrom: pink
+colorTo: purple
 sdk: streamlit
 sdk_version: 1.10.0
 app_file: videogenic.py

videogenic.py

@@ -30,6 +30,18 @@ from moviepy.video.io.ffmpeg_tools import ffmpeg_extract_subclip
 from moviepy.editor import *
 import glob
 
+@st.cache(show_spinner=False)
+def load_mode():
+  device = 'cuda' if torch.cuda.is_available() else 'cpu'
+  model, preprocess = openai_clip.load('ViT-B/32', device=device)
+  if 'model' not in st.session_state:
+    st.session_state.model = model
+    st.session_state.preprocess = preprocess
+    st.session_state.device = device
+  st.session_state.model = model
+  st.session_state.preprocess = preprocess
+  st.session_state.device = device
+
 def fetch_video(url):
   yt = YouTube(url)
   streams = yt.streams.filter(adaptive=True, subtype='mp4', resolution='360p', only_video=True)
@@ -490,15 +502,7 @@ if 'progress' not in st.session_state:
 
 if st.session_state.progress == 1:
   with st.spinner('Loading model...'):
-    device = 'cuda' if torch.cuda.is_available() else 'cpu'
-    model, preprocess = openai_clip.load('ViT-B/32', device=device)
-    if 'model' not in st.session_state:
-      st.session_state.model = model
-      st.session_state.preprocess = preprocess
-      st.session_state.device = device
-    st.session_state.model = model
-    st.session_state.preprocess = preprocess
-    st.session_state.device = device
+    load_model()
   domain = st.selectbox('Select video',('Skydiving', 'Surfing')) # Entire journey, montage, vlog
   if 'domain' not in st.session_state:
     st.session_state.domain = domain
@@ -591,10 +595,11 @@ if st.session_state.progress == 2:
     if st.button('Generate video'):
       with st.spinner('Generating highlight video...'):
         edit_video(template, st.session_state.df_all)
+        st.balloons()
   elif st.session_state.mode == 'User selection':
     if st.button('Generate video'):
       left = st.session_state.selection['x'][0]
       right = st.session_state.selection['x'][1]
       with st.spinner('Generating highlight video...'):
         crop_video(st.session_state.df_all, left, right)
-  st.balloons()
+        st.balloons()