David Chuan-En Lin commited on
Commit
a6f9eca
·
1 Parent(s): d2e201b
Files changed (1) hide show
  1. videogenic.py +1 -1
videogenic.py CHANGED
@@ -31,7 +31,7 @@ from moviepy.editor import *
31
  import glob
32
 
33
  @st.cache(show_spinner=False)
34
- def load_mode():
35
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
36
  model, preprocess = openai_clip.load('ViT-B/32', device=device)
37
  if 'model' not in st.session_state:
 
31
  import glob
32
 
33
  @st.cache(show_spinner=False)
34
+ def load_model():
35
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
36
  model, preprocess = openai_clip.load('ViT-B/32', device=device)
37
  if 'model' not in st.session_state: