Update custom model files, README, and requirements
Browse files- asr_pipeline.py +0 -3
asr_pipeline.py
CHANGED
|
@@ -33,12 +33,9 @@ class ASRPipeline(transformers.AutomaticSpeechRecognitionPipeline):
|
|
| 33 |
"sampling_rate": inputs.get("sampling_rate", self.feature_extractor.sampling_rate),
|
| 34 |
}
|
| 35 |
|
| 36 |
-
print(f"DEBUG preprocess called, preprocess_params={preprocess_params}")
|
| 37 |
for item in super().preprocess(inputs, **preprocess_params):
|
| 38 |
-
print(f"DEBUG item from super: {item.keys()}")
|
| 39 |
if "is_last" not in item:
|
| 40 |
item["is_last"] = True
|
| 41 |
-
print(f"DEBUG yielding item with keys: {item.keys()}")
|
| 42 |
yield item
|
| 43 |
|
| 44 |
def _forward(self, model_inputs, **generate_kwargs) -> dict[str, Any]:
|
|
|
|
| 33 |
"sampling_rate": inputs.get("sampling_rate", self.feature_extractor.sampling_rate),
|
| 34 |
}
|
| 35 |
|
|
|
|
| 36 |
for item in super().preprocess(inputs, **preprocess_params):
|
|
|
|
| 37 |
if "is_last" not in item:
|
| 38 |
item["is_last"] = True
|
|
|
|
| 39 |
yield item
|
| 40 |
|
| 41 |
def _forward(self, model_inputs, **generate_kwargs) -> dict[str, Any]:
|