shixian.shi
2023-03-03 a1447d12cc7b18a260a4d1cd8ff572f8e78eaba4
funasr/runtime/python/onnxruntime/rapid_paraformer/paraformer_onnx.py
@@ -24,7 +24,7 @@
    def __init__(self, model_dir: Union[str, Path] = None,
                 batch_size: int = 1,
                 device_id: Union[str, int] = "-1",
                 plot_timestamp: bool = True,
                 plot_timestamp: bool = False,
                 ):
        if not Path(model_dir).exists():
@@ -96,7 +96,7 @@
            x_adj = 0.045 if char != '<sil>' else 0.12
            ax1.text((start + end) * 0.5 - x_adj, 0, char)
        # plt.legend()
        plotname = "debug.png"
        plotname = "funasr/runtime/python/onnxruntime/debug.png"
        plt.savefig(plotname, bbox_inches='tight')
    def load_data(self,