Upload 5 files

.gitattributes

@@ -25,7 +25,6 @@
 *.safetensors filter=lfs diff=lfs merge=lfs -text
 saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.tar.* filter=lfs diff=lfs merge=lfs -text
-*.tar filter=lfs diff=lfs merge=lfs -text
 *.tflite filter=lfs diff=lfs merge=lfs -text
 *.tgz filter=lfs diff=lfs merge=lfs -text
 *.wasm filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1,158 @@
+# Created by .ignore support plugin (hsz.mobi)
+### Python template
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+.pytest_cache
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+pip-wheel-metadata/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+# *.manifest
+# *.spec
+*.res
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+.python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+#Pipfile.lock
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+#idea
+.vs
+.vscode
+.idea
+/images
+/models
+
+#models
+*.onnx
+
+*.ttf
+*.ttc
+
+long1.jpg
+
+*.bin
+*.mapping
+*.xml
+
+*.pdiparams
+*.pdiparams.info
+*.pdmodel
+
+.DS_Store

app.py

@@ -0,0 +1,115 @@
+# -*- encoding: utf-8 -*-
+# @Author: SWHL
+# @Contact: [email protected]
+from pathlib import Path
+
+import numpy as np
+import streamlit as st
+from PIL import Image
+from rapid_latex_ocr import LatexOCR
+from streamlit_cropper import st_cropper
+from streamlit_image_select import image_select
+
+st.set_option("deprecation.showfileUploaderEncoding", False)
+
+
+class RecEquation:
+    def __init__(self, model_dir: str):
+        model_dir = Path(model_dir)
+
+        image_resizer_path = model_dir / "image_resizer.onnx"
+        encoder_path = model_dir / "encoder.onnx"
+        decoder_path = model_dir / "decoder.onnx"
+        tokenizer_json = model_dir / "tokenizer.json"
+        self.model = LatexOCR(
+            image_resizer_path=str(image_resizer_path),
+            encoder_path=str(encoder_path),
+            decoder_path=str(decoder_path),
+            tokenizer_json=str(tokenizer_json),
+        )
+
+    def __call__(self, img: np.ndarray):
+        result, elapse = self.model(img)
+        return result, elapse
+
+
+if __name__ == "__main__":
+    st.markdown(
+        "<h1 style='text-align: center;'><a href='https://github.com/RapidAI/RapidLatexOCR' style='text-decoration: none'>Rapid Latex OCR</a></h1>",
+        unsafe_allow_html=True,
+    )
+    st.markdown(
+        """
+    <p align="left">
+        <a href=""><img src="https://img.shields.io/badge/Python->=3.6,<3.12-aff.svg"></a>
+        <a href=""><img src="https://img.shields.io/badge/OS-Linux%2C%20Win%2C%20Mac-pink.svg"></a>
+        <a href="https://pepy.tech/project/rapid_latex_ocr"><img src="https://static.pepy.tech/personalized-badge/rapid_latex_ocr?period=total&units=abbreviation&left_color=grey&right_color=blue&left_text=Downloads"></a>
+        <a href="https://pypi.org/project/rapid_latex_ocr/"><img alt="PyPI" src="https://img.shields.io/pypi/v/rapid_latex_ocr"></a>
+        <a href="https://semver.org/"><img alt="SemVer2.0" src="https://img.shields.io/badge/SemVer-2.0-brightgreen"></a>
+        <a href="https://github.com/psf/black"><img src="https://img.shields.io/badge/code%20style-black-000000.svg"></a>
+    </p>
+    """,
+        unsafe_allow_html=True,
+    )
+
+    # Upload an image and set some options for demo purposes
+    img_file = st.sidebar.file_uploader(label="Upload a file", type=["png", "jpg"])
+    realtime_update = st.sidebar.checkbox(label="Update in Real Time", value=False)
+    box_color = st.sidebar.color_picker(label="Box Color", value="#0000FF")
+    aspect_choice = st.sidebar.radio(
+        label="Aspect Ratio", options=["Free", "1:1", "16:9", "4:3", "2:3"]
+    )
+    aspect_dict = {
+        "Free": None,
+        "1:1": (1, 1),
+        "16:9": (16, 9),
+        "4:3": (4, 3),
+        "2:3": (2, 3),
+    }
+    aspect_ratio = aspect_dict[aspect_choice]
+    with st.sidebar.container():
+        img = image_select(
+            label="Examples（click to select）:",
+            images=[
+                "images/equation.png",
+                "images/eq_2.png",
+                "images/eq_3.png",
+                "images/eq_4.png",
+            ],
+            key="equation_default",
+            use_container_width=False,
+        )
+
+    rec_eq_sys = RecEquation(model_dir="models")
+
+    select_img_container = st.container()
+
+    st.markdown("#### Select image:")
+    img_empty = st.empty()
+
+    img_empty.image(img, use_column_width=False)
+    rec_res, elapse = rec_eq_sys(img)
+
+    if img_file:
+        img = Image.open(img_file)
+
+        # Get a cropped image from the frontend
+        with select_img_container:
+            if not realtime_update:
+                select_img_container.markdown("#### Double click to save crop")
+
+            img = st_cropper(
+                img,
+                realtime_update=realtime_update,
+                box_color=box_color,
+                aspect_ratio=aspect_ratio,
+            )
+
+        img_empty.image(img, use_column_width=False)
+        rec_res, elapse = rec_eq_sys(np.array(img))
+
+    st.markdown(f"#### Rec result (cost: {elapse:.4f}s):")
+    st.latex(rec_res)
+
+    st.markdown("#### Latex source code:")
+    st.code(rec_res)

requirements.txt

@@ -0,0 +1,6 @@
+altair<5
+Pillow
+numpy
+rapid_latex_ocr
+streamlit_image_select
+streamlit_cropper