Update app.py
Browse files
app.py
CHANGED
@@ -145,7 +145,7 @@ def load_file(fname):
|
|
145 |
wcs = WCS(hdul[0].header)
|
146 |
return data, wcs
|
147 |
|
148 |
-
@st.cache #_resource
|
149 |
def load_CADET():
|
150 |
model = from_pretrained_keras("Plsek/CADET-v1")
|
151 |
# model = load_model("CADET.hdf5")
|
|
|
145 |
wcs = WCS(hdul[0].header)
|
146 |
return data, wcs
|
147 |
|
148 |
+
@st.cache(allow_output_mutation=True) #_resource
|
149 |
def load_CADET():
|
150 |
model = from_pretrained_keras("Plsek/CADET-v1")
|
151 |
# model = load_model("CADET.hdf5")
|