diff --git a/speechserving/speechserving/engine/asr/python/asr_engine.py b/speechserving/speechserving/engine/asr/python/asr_engine.py index e8289332..a18f906a 100644 --- a/speechserving/speechserving/engine/asr/python/asr_engine.py +++ b/speechserving/speechserving/engine/asr/python/asr_engine.py @@ -81,7 +81,6 @@ class ASRServerExecutor(ASRExecutor): """ audio_file = input - # logger.info("Preprocess audio_file:" + audio_file) # Get the object for feature extraction if "deepspeech2online" in model_type or "deepspeech2offline" in model_type: diff --git a/speechserving/speechserving/engine/engine_factory.py b/speechserving/speechserving/engine/engine_factory.py index 2b9f9db7..bc0c4565 100644 --- a/speechserving/speechserving/engine/engine_factory.py +++ b/speechserving/speechserving/engine/engine_factory.py @@ -11,15 +11,18 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. +from typing import Text + from engine.asr.python.asr_engine import ASREngine from engine.tts.python.tts_engine import TTSEngine + __all__ = ['EngineFactory'] class EngineFactory(object): @staticmethod - def get_engine(engine_name: str): + def get_engine(engine_name: Text): if engine_name == 'asr': return ASREngine() elif engine_name == 'tts': diff --git a/speechserving/speechserving/restful/asr_api.py b/speechserving/speechserving/restful/asr_api.py index 6ac647bc..c63cd76c 100644 --- a/speechserving/speechserving/restful/asr_api.py +++ b/speechserving/speechserving/restful/asr_api.py @@ -42,7 +42,7 @@ def help(): "global": "success" }, "result": { - "description": "tts server", + "description": "asr server", "input": "base64 string of wavfile", "output": "transcription" } diff --git a/speechserving/tests/16_audio.wav b/speechserving/tests/16_audio.wav new file mode 100644 index 00000000..3cfa5074 Binary files /dev/null and b/speechserving/tests/16_audio.wav differ diff --git a/speechserving/tests/http_client.py b/speechserving/tests/http_client.py index 73f5c18d..14adb574 100644 --- a/speechserving/tests/http_client.py +++ b/speechserving/tests/http_client.py @@ -14,10 +14,8 @@ import requests import json import time import base64 -import soundfile import io -import argparse def readwav2base64(wav_file): """ @@ -29,7 +27,7 @@ def readwav2base64(wav_file): return base64_string -def main(args): +def main(): """ main func """ @@ -58,9 +56,4 @@ def main(args): if __name__ == "__main__": - parser = argparse.ArgumentParser() - parser.add_argument("--model_type", action="store", - help="model type: u2, dp2", default="dp2") - args = parser.parse_args() - - main(args) + main()