Spaces:
Runtime error
Runtime error
David Chuan-En Lin
commited on
Commit
·
a6f9eca
1
Parent(s):
d2e201b
Caching
Browse files- 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
|
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:
|