Update app.py

app.py

@@ -19,8 +19,8 @@ class Speech2Text:
 
     def feature2logits(self, features):
         with torch.no_grad():
-            return model(features.input_values.to(device),
-                         attention_mask=features.attention_mask.to(device)).logits.numpy()[0]
+            return wav2vec_model(features.input_values.to(device),
+                                 attention_mask=features.attention_mask.to(device)).logits.numpy()[0]
 
     def __call__(self, path):
         logits = self.feature2logits(self.wav2feature(path))