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="🧮")