update

api/telos_api.py

@@ -3,17 +3,8 @@
 from fastapi import FastAPI, File, UploadFile
 from fastapi.responses import FileResponse, JSONResponse
 
-from telos import (
-    CRNN,
-    OCR,
-    DBNet,
-    DetFormula,
-    LatexOCR,
-    Layout,
-    ReadingOrder,
-    Table_TSR,
-    YOLOv8,
-)
+from telos import (CRNN, OCR, DBNet, DetFormula, LatexOCR, Layout,
+                   ReadingOrder, Table_TSR, YOLOv8)
 
 app = FastAPI()
 

telos/__init__.py

@@ -4,9 +4,9 @@
 __project__ = "telos"
 
 from telos.config import *
-
 # from telos.utils import check_source
-from telos.core import CRNN, CVModel, DBNet, LatexOCR, Lore, ReadingOrder, YOLOv8
+from telos.core import (CRNN, CVModel, DBNet, LatexOCR, Lore, ReadingOrder,
+                        YOLOv8)
 from telos.data import IMG_FORMATS, MetaFile
 from telos.modules import OCR, DetFormula, Layout, Table_TSR
 

telos/config/__init__.py

@@ -8,7 +8,6 @@
 
 from dynaconf import Dynaconf
 
-
 _base_dir = Path(__file__).parent.parent
 
 _settings_files = [
@@ -34,8 +33,8 @@
 )
 
 
-from .settings1 import __curr_path__, __model_path__
 from .settings1 import *  # formula_model, formula_labels
+from .settings1 import __curr_path__, __model_path__
 
 # __all__ = (
 #     "__curr_path__",

telos/core/__init__.py

@@ -1,6 +1,5 @@
 from .base import CVModel, OrtInferSession
 from .detection import DBNet, Lore, YOLOv8
-
 # from .imageprocess import *
 from .reading_order import ReadingOrder
 from .recognition import CRNN, LatexOCR

telos/core/base/model.py

@@ -8,7 +8,8 @@
 import numpy as np
 import onnxruntime as ort
 from huggingface_hub import snapshot_download
-from onnxruntime import GraphOptimizationLevel, InferenceSession, SessionOptions
+from onnxruntime import (GraphOptimizationLevel, InferenceSession,
+                         SessionOptions)
 
 from telos import __project__
 from telos.config import __model_path__

telos/core/recognition/latexocr.py

@@ -9,7 +9,8 @@
 import cv2
 import numpy as np
 import yaml
-from onnxruntime import GraphOptimizationLevel, InferenceSession, SessionOptions
+from onnxruntime import (GraphOptimizationLevel, InferenceSession,
+                         SessionOptions)
 from PIL import Image, UnidentifiedImageError
 from tokenizers import Tokenizer
 from tokenizers.models import BPE

telos/data/__init__.py

@@ -1,4 +1,5 @@
-from .types import IMG_FORMATS, MetaArea, MetaBbox, MetaFile, MetaLayout, MetaText
+from .types import (IMG_FORMATS, MetaArea, MetaBbox, MetaFile, MetaLayout,
+                    MetaText)
 
 __all__ = [
     "IMG_FORMATS",

webui/demo.py

@@ -6,17 +6,8 @@
 import psutil
 import streamlit as st
 
-from telos import (
-    CRNN,
-    OCR,
-    DBNet,
-    DetFormula,
-    LatexOCR,
-    Layout,
-    ReadingOrder,
-    Table_TSR,
-    YOLOv8,
-)
+from telos import (CRNN, OCR, DBNet, DetFormula, LatexOCR, Layout,
+                   ReadingOrder, Table_TSR, YOLOv8)
 
 st.set_page_config(layout="wide")
 st.title("Telos 演示")