Update custom_interface.py
Browse files- custom_interface.py +1 -3
custom_interface.py
CHANGED
|
@@ -24,10 +24,8 @@ class ASR(Pretrained):
|
|
| 24 |
prediction = []
|
| 25 |
for sent in predicted_words:
|
| 26 |
sent = self.filter_repetitions(sent, 3)
|
| 27 |
-
# sent = " ".join(sent)
|
| 28 |
prediction.append(sent)
|
| 29 |
-
|
| 30 |
-
predicted_words = [prediction]
|
| 31 |
return predicted_words
|
| 32 |
|
| 33 |
def filter_repetitions(self, seq, max_repetition_length):
|
|
|
|
| 24 |
prediction = []
|
| 25 |
for sent in predicted_words:
|
| 26 |
sent = self.filter_repetitions(sent, 3)
|
|
|
|
| 27 |
prediction.append(sent)
|
| 28 |
+
predicted_words = prediction
|
|
|
|
| 29 |
return predicted_words
|
| 30 |
|
| 31 |
def filter_repetitions(self, seq, max_repetition_length):
|