From 110cb29d6c951207b57e3e755feb5c4f58977d0d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E4=B8=89=E6=B4=8B=E4=B8=89=E6=B4=8B?= <1258009915@qq.com> Date: Sat, 1 Mar 2025 23:37:06 +0800 Subject: [PATCH] [fix] Add project prefix --- texteller/server.py | 10 +++++----- texteller/web.py | 14 +++++++------- 2 files changed, 12 insertions(+), 12 deletions(-) diff --git a/texteller/server.py b/texteller/server.py index e17db20..d22a706 100644 --- a/texteller/server.py +++ b/texteller/server.py @@ -11,11 +11,11 @@ from ray import serve from ray.serve.handle import DeploymentHandle from onnxruntime import InferenceSession -from models.ocr_model.utils.inference import inference as rec_inference -from models.det_model.inference import predict as det_inference -from models.ocr_model.model.TexTeller import TexTeller -from models.det_model.inference import PredictConfig -from models.ocr_model.utils.to_katex import to_katex +from texteller.models.ocr_model.utils.inference import inference as rec_inference +from texteller.models.det_model.inference import predict as det_inference +from texteller.models.ocr_model.model.TexTeller import TexTeller +from texteller.models.det_model.inference import PredictConfig +from texteller.models.ocr_model.utils.to_katex import to_katex PYTHON_VERSION = str(sys.version_info.major) + '.' + str(sys.version_info.minor) diff --git a/texteller/web.py b/texteller/web.py index 7a497d9..ff13a74 100644 --- a/texteller/web.py +++ b/texteller/web.py @@ -9,15 +9,15 @@ import streamlit as st from PIL import Image from streamlit_paste_button import paste_image_button as pbutton from onnxruntime import InferenceSession -from models.thrid_party.paddleocr.infer import predict_det, predict_rec -from models.thrid_party.paddleocr.infer import utility +from texteller.models.thrid_party.paddleocr.infer import predict_det, predict_rec +from texteller.models.thrid_party.paddleocr.infer import utility -from models.utils import mix_inference -from models.det_model.inference import PredictConfig +from texteller.models.utils import mix_inference +from texteller.models.det_model.inference import PredictConfig -from models.ocr_model.model.TexTeller import TexTeller -from models.ocr_model.utils.inference import inference as latex_recognition -from models.ocr_model.utils.to_katex import to_katex +from texteller.models.ocr_model.model.TexTeller import TexTeller +from texteller.models.ocr_model.utils.inference import inference as latex_recognition +from texteller.models.ocr_model.utils.to_katex import to_katex st.set_page_config(page_title="TexTeller", page_icon="🧮")