Fix deprecated class call

jollyjonson · Dec 11, 2023 · a5e90f5 · a5e90f5
1 parent fc29d1c
commit a5e90f5
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/frechet_audio_distance/frechet_audio_distance.py b/frechet_audio_distance/frechet_audio_distance.py
@@ -407,9 +407,7 @@ def _build_vggish_as_keras_model():
         x = tf.keras.layers.Flatten()(x)
         x = tf.keras.layers.Dense(4096, activation="relu")(x)
         x = tf.keras.layers.Dense(4096, activation="relu")(x)
-        x = tf.keras.layers.Dense(
-            VGGishParams.EMBEDDING_SIZE, activation=None
-        )(x)
+        x = tf.keras.layers.Dense(VGGish.embedding_size, activation=None)(x)
         embedding = tf.identity(x, name="embedding")
         return tf.keras.Model(inputs=[input], outputs=[embedding])