From 0cf7339171ee136721059c3c185b5d92d160b17a Mon Sep 17 00:00:00 2001
From: zhaomingwork <61895407+zhaomingwork@users.noreply.github.com>
Date: 星期一, 21 八月 2023 14:43:07 +0800
Subject: [PATCH] add hotwords for h5 and java (#876)
---
funasr/runtime/java/FunasrWsClient.java | 19 ++++++++++++++++++-
1 files changed, 18 insertions(+), 1 deletions(-)
diff --git a/funasr/runtime/java/FunasrWsClient.java b/funasr/runtime/java/FunasrWsClient.java
index 43407f3..74a8695 100644
--- a/funasr/runtime/java/FunasrWsClient.java
+++ b/funasr/runtime/java/FunasrWsClient.java
@@ -79,6 +79,10 @@
obj.put("chunk_size", array);
obj.put("chunk_interval", new Integer(chunkInterval));
obj.put("wav_name", wavName);
+ if(FunasrWsClient.hotwords.trim().length()>0)
+ {
+ obj.put("hotwords", FunasrWsClient.hotwords.trim());
+ }
if(suffix.equals("wav")){
suffix="pcm";
}
@@ -185,6 +189,10 @@
try {
jsonObject = (JSONObject) jsonParser.parse(message);
logger.info("text: " + jsonObject.get("text"));
+ if(jsonObject.containsKey("timestamp"))
+ {
+ logger.info("timestamp: " + jsonObject.get("timestamp"));
+ }
} catch (org.json.simple.parser.ParseException e) {
e.printStackTrace();
}
@@ -222,6 +230,7 @@
static String strChunkSize = "5,10,5";
static int chunkInterval = 10;
static int sendChunkSize = 1920;
+ static String hotwords="";
String wavName = "javatest";
@@ -270,6 +279,12 @@
.setDefault("offline")
.type(String.class)
.required(false);
+ parser
+ .addArgument("--hotwords")
+ .help("hotwords, splited by space")
+ .setDefault("")
+ .type(String.class)
+ .required(false);
String srvIp = "";
String srvPort = "";
String wavPath = "";
@@ -277,7 +292,7 @@
String chunk_size = "";
int chunk_interval = 10;
String strmode = "offline";
-
+ String hot="";
try {
Namespace ns = parser.parseArgs(args);
srvIp = ns.get("host");
@@ -287,6 +302,7 @@
chunk_size = ns.get("chunk_size");
chunk_interval = ns.get("chunk_interval");
strmode = ns.get("mode");
+ hot=ns.get("hotwords");
System.out.println(srvPort);
} catch (ArgumentParserException ex) {
@@ -298,6 +314,7 @@
FunasrWsClient.chunkInterval = chunk_interval;
FunasrWsClient.wavPath = wavPath;
FunasrWsClient.mode = strmode;
+ FunasrWsClient.hotwords=hot;
System.out.println(
"serIp="
+ srvIp
--
Gitblit v1.9.1