apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-16 05:30:58 +08:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

Add uie python api (#214)

Browse Source 
* add uie pybind

* Add uie result pybind

* Add uie python class

* fix UIEModel pythonargs

* Add schema node pybind

* remove uie print

* Fix cpp build ci
This commit is contained in:
Jack Zhou
2022-09-13 19:03:06 +08:00
committed by GitHub
parent 82580ac11e
commit 54bea3160d
11 changed files with 272 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
csrc/fastdeploy/text/text_pybind.cc
View File

@@ -10,4 +10,25 @@
// distributed under the License is distributed on an "AS IS" BASIS,
// 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.
// limitations under the License.
#include "fastdeploy/pybind/main.h"
namespace fastdeploy {
void BindUIE(pybind11::module& m);
void BindText(pybind11::module& m) {
pybind11::class_<text::UIEResult>(m, "UIEResult")
.def(pybind11::init())
.def_readwrite("start", &text::UIEResult::start_)
.def_readwrite("end", &text::UIEResult::end_)
.def_readwrite("probability_", &text::UIEResult::probability_)
.def_readwrite("text", &text::UIEResult::text_)
.def_readwrite("relation", &text::UIEResult::relation_)
.def("__repr__", &text::UIEResult::Str)
.def("__str__", &text::UIEResult::Str);
BindUIE(m);
}
} // namespace fastdeploy