Update app

neural_acoustic_distance.py

@@ -28,10 +28,9 @@ if model_id == "other":
     model_id = st.text_input("Enter the wav2vec 2.0 model you want to use:", value = "facebook/wav2vec2-large-960h", key = "model")
 
 try:
-    cfg = AutoConfig.from_pretrained(model_id)
-    print(cfg.num_hidden_layers)
+    # cfg = AutoConfig.from_pretrained(model_id)
     layer = st.number_input("Select the layer you want to use:",
-        min_value = 1, max_value = cfg.num_hidden_layers, value=10)
+        min_value = 1, max_value = 24, value=10)
 
     def load_wav2vec2_featurizer(model_id: str, layer: Optional[int] = None):
         from transformers.models.wav2vec2 import Wav2Vec2Model