Update videogenic.py

videogenic.py

@@ -516,8 +516,8 @@ if st.session_state.progress == 1:
         st.session_state.video = video_bytes
       st.session_state.video = video_bytes
       # st.video(st.session_state.video)
-      # video_frames, fps, x_dim, y_dim = video_to_frames(video_name) # first run; video_to_info
-      # np.save(f'files/{st.session_state.domain.lower()}.npy', video_frames)
+      video_frames, fps, x_dim, y_dim = video_to_frames(video_name) # first run; video_to_info
+      np.save(f'files/{st.session_state.domain.lower()}.npy', video_frames)
       fps, x_dim, y_dim = video_to_info(video_name)
       video_frames = np.load(f'files/{st.session_state.domain.lower()}.npy', allow_pickle=True)
       if 'video_frames' not in st.session_state:
@@ -530,8 +530,8 @@ if st.session_state.progress == 1:
       st.session_state.x_dim = x_dim
       st.session_state.y_dim = y_dim
       print('Extracted frames')
-      # encoded_frames = encode_frames(video_frames) # first run
-      # np.save(f'files/{st.session_state.domain.lower()}_features.npy', encoded_frames)
+      encoded_frames = encode_frames(video_frames) # first run
+      np.save(f'files/{st.session_state.domain.lower()}_features.npy', encoded_frames)
       encoded_frames = np.load(f'files/{st.session_state.domain.lower()}_features.npy', allow_pickle=True)
       if 'video_features' not in st.session_state:
         # st.session_state.video_features = encoded_frames