Daniel Verdu commited on
Commit
6fca209
1 Parent(s): b0c41bf

updated loading model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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'