Spaces:
Runtime error
Runtime error
Daniel Verdu
commited on
Commit
•
6fca209
1
Parent(s):
b0c41bf
updated loading model
Browse files
app.py
CHANGED
@@ -34,7 +34,7 @@ model_folder = 'models/'
|
|
34 |
max_img_size = 800
|
35 |
################################
|
36 |
|
37 |
-
@st.cache(allow_output_mutation=True)
|
38 |
def load_model(model_dir, option):
|
39 |
if option.lower() == 'artistic':
|
40 |
model_url = 'https://data.deepai.org/deoldify/ColorizeArtistic_gen.pth'
|
|
|
34 |
max_img_size = 800
|
35 |
################################
|
36 |
|
37 |
+
# @st.cache(allow_output_mutation=True)
|
38 |
def load_model(model_dir, option):
|
39 |
if option.lower() == 'artistic':
|
40 |
model_url = 'https://data.deepai.org/deoldify/ColorizeArtistic_gen.pth'
|