Hugging Face's logo

Spaces:
Ashrafb
/
Bwcdocker
Running

App Files Community
ArchitSharma commited on
Commit
6dbedbe
·
1 Parent(s): a408126

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ from src.app_utils import get_model_bin
23
  device.set(device=DeviceId.CPU)
24
 
25
 
26
- @st.cache(allow_output_mutation=True, show_spinner=False)
27
  def load_model(model_dir, option):
28
  if option.lower() == 'artistic':
29
  model_url = 'https://data.deepai.org/deoldify/ColorizeArtistic_gen.pth'
 
23
  device.set(device=DeviceId.CPU)
24
 
25
 
26
+ @st.cache_resource
27
  def load_model(model_dir, option):
28
  if option.lower() == 'artistic':
29
  model_url = 'https://data.deepai.org/deoldify/ColorizeArtistic_gen.pth'