Update neural_acoustic_distance.py

neural_acoustic_distance.py

@@ -39,7 +39,7 @@ def compute_costs(gcm):
     return res, n
 
 
-# @st.cache(show_spinner=False, hash_funcs={torch.nn.parameter.Parameter: lambda _: None})
+@st.cache(show_spinner=False, hash_funcs={torch.nn.parameter.Parameter: lambda _: None}, max_entries=1)
 def load_wav2vec2_featurizer(model_id: str, layer: Optional[int] = None):
     transformers.logging.set_verbosity(transformers.logging.ERROR)
 
@@ -86,7 +86,7 @@ def load_wav2vec2_featurizer(model_id: str, layer: Optional[int] = None):
     return _featurize
 
 
-# @st.cache(persist=True, show_spinner=False)
+@st.cache(persist=True, show_spinner=False, max_entries=3)
 def run(model_id, layer, filename_x, filename_y):
     featurizer = load_wav2vec2_featurizer(model_id, layer)