call supers

README.md

@@ -1,6 +1,6 @@
 ---
 title: Speech analysis
-emoji: 🎤
+emoji:
 colorFrom: gray
 colorTo: gray
 sdk: gradio

app.py

@@ -89,15 +89,13 @@ def _forward(
         hidden_states, mask_time_indices=mask_time_indices, attention_mask=attention_mask
     )
 
-    encoder_outputs = self.encoder(
+    hidden_states = self.wav2vec2.encoder(
         hidden_states,
         attention_mask=attention_mask,
         output_attentions=output_attentions,
         output_hidden_states=output_hidden_states,
         return_dict=return_dict,
-    )
-
-    hidden_states = encoder_outputs[0]
+    )[0]
 
     if self.adapter is not None:
         raise ValueError
@@ -111,29 +109,27 @@ def _forward_and_cnn7(
     input_values,
     attention_mask=None):
 
-    frozen_cnn7 = self.feature_extractor(input_values)
+    frozen_cnn7 = self.wav2vec2.feature_extractor(input_values)
     frozen_cnn7 = frozen_cnn7.transpose(1, 2)
 
     if attention_mask is not None:
         # compute reduced attention_mask corresponding to feature vectors
-        attention_mask = self._get_feature_vector_attention_mask(
+        attention_mask = self.wav2vec2._get_feature_vector_attention_mask(
             frozen_cnn7.shape[1], attention_mask, add_adapter=False
         )
 
-    hidden_states, _ = self.feature_projection(frozen_cnn7)  # grad=True non frozen
-    hidden_states = self._mask_hidden_states(
+    hidden_states, _ = self.wav2vec2.feature_projection(frozen_cnn7)  # grad=True non frozen
+    hidden_states = self.wav2vec2._mask_hidden_states(
         hidden_states, mask_time_indices=mask_time_indices, attention_mask=attention_mask
     )
 
-    encoder_outputs = self.encoder(
+    hidden_states = self.wav2vec2.encoder(
         hidden_states,
         attention_mask=attention_mask,
         output_attentions=output_attentions,
         output_hidden_states=output_hidden_states,
         return_dict=return_dict,
-    )
-
-    hidden_states = encoder_outputs[0]
+    )[0]
 
     if self.adapter is not None:
         raise ValueError