Spaces:

ArchitSharma
/

Digital-Photo-Color-Restoration

Running

ArchitSharma commited on Apr 16, 2023

Commit

6dbedbe

1 Parent(s): a408126

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -23,7 +23,7 @@ from src.app_utils import get_model_bin
 device.set(device=DeviceId.CPU)
-@st.cache(allow_output_mutation=True, show_spinner=False)
 def load_model(model_dir, option):
     if option.lower() == 'artistic':
         model_url = 'https://data.deepai.org/deoldify/ColorizeArtistic_gen.pth'

 device.set(device=DeviceId.CPU)
+@st.cache_resource
 def load_model(model_dir, option):
     if option.lower() == 'artistic':
         model_url = 'https://data.deepai.org/deoldify/ColorizeArtistic_gen.pth'