mirror of
https://github.com/modelscope/FunASR
synced 2025-09-15 14:48:36 +08:00
add example of grpc client
This commit is contained in:
parent
7687f64729
commit
fb06c50b11
@ -55,3 +55,63 @@ Usage: ./cmake/build/paraformer_server port thread_num /path/to/model_file quant
|
|||||||
cd ../python/grpc
|
cd ../python/grpc
|
||||||
python grpc_main_client_mic.py --host $server_ip --port 10108
|
python grpc_main_client_mic.py --host $server_ip --port 10108
|
||||||
```
|
```
|
||||||
|
The `grpc_main_client_mic.py` follows the [original design] (https://github.com/alibaba-damo-academy/FunASR/tree/main/funasr/runtime/python/grpc#workflow-in-desgin) by sending audio_data with chunks. If you want to send audio_data in one request, here is an example:
|
||||||
|
|
||||||
|
```
|
||||||
|
# go to ../python/grpc to find this package
|
||||||
|
import paraformer_pb2
|
||||||
|
|
||||||
|
|
||||||
|
class RecognizeStub:
|
||||||
|
def __init__(self, channel):
|
||||||
|
self.Recognize = channel.stream_stream(
|
||||||
|
'/paraformer.ASR/Recognize',
|
||||||
|
request_serializer=paraformer_pb2.Request.SerializeToString,
|
||||||
|
response_deserializer=paraformer_pb2.Response.FromString,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
async def send(channel, data, speaking, isEnd):
|
||||||
|
stub = RecognizeStub(channel)
|
||||||
|
req = paraformer_pb2.Request()
|
||||||
|
if data:
|
||||||
|
req.audio_data = data
|
||||||
|
req.user = 'zz'
|
||||||
|
req.language = 'zh-CN'
|
||||||
|
req.speaking = speaking
|
||||||
|
req.isEnd = isEnd
|
||||||
|
q = queue.SimpleQueue()
|
||||||
|
q.put(req)
|
||||||
|
return stub.Recognize(iter(q.get, None))
|
||||||
|
|
||||||
|
# send the audio data once
|
||||||
|
async def grpc_rec(data, grpc_uri):
|
||||||
|
with grpc.insecure_channel(grpc_uri) as channel:
|
||||||
|
b = time.time()
|
||||||
|
response = await send(channel, data, False, False)
|
||||||
|
resp = response.next()
|
||||||
|
text = ''
|
||||||
|
if 'decoding' == resp.action:
|
||||||
|
resp = response.next()
|
||||||
|
if 'finish' == resp.action:
|
||||||
|
text = json.loads(resp.sentence)['text']
|
||||||
|
response = await send(channel, None, False, True)
|
||||||
|
return {
|
||||||
|
'text': text,
|
||||||
|
'time': time.time() - b,
|
||||||
|
}
|
||||||
|
|
||||||
|
async def test():
|
||||||
|
# fc = FunAsrGrpcClient('127.0.0.1', 9900)
|
||||||
|
# t = await fc.rec(wav.tobytes())
|
||||||
|
# print(t)
|
||||||
|
wav, _ = sf.read('z-10s.wav', dtype='int16')
|
||||||
|
uri = '127.0.0.1:9900'
|
||||||
|
res = await grpc_rec(wav.tobytes(), uri)
|
||||||
|
print(res)
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
asyncio.run(test())
|
||||||
|
|
||||||
|
```
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user