Merge pull request #60 from OleehyO/pre_release

Pre release

README.md

@@ -169,7 +169,8 @@ Go to the `src/` directory and run the following command:
 Enter `http://localhost:8501` in a browser to view the web demo.
 
 > [!NOTE]
-> If you are Windows user, please run the `start_web.bat` file instead.
+> 1. For Windows users, please run the `start_web.bat` file.
+> 2. When using onnxruntime + GPU for inference, you need to install onnxruntime-gpu.
 
 ## 🔍 Formula Detection
 

assets/README_zh.md

@@ -204,7 +204,8 @@ TexTeller默认使用轻量的[PaddleOCR](https://github.com/PaddlePaddle/Paddle
 在浏览器里输入 `http://localhost:8501`就可以看到web demo
 
 > [!NOTE]
-> 对于Windows用户, 请运行 `start_web.bat`文件。
+> 1. 对于Windows用户, 请运行 `start_web.bat`文件。
+> 2. 使用onnxruntime + gpu 推理时，需要安装onnxruntime-gpu
 
 ## 🔍 公式检测
 

setup.py

@@ -1,11 +1,10 @@
 from setuptools import setup, find_packages
-import platform
+
 
 # Define the base dependencies
 install_requires = [
     "torch",
     "torchvision",
-    "torchaudio",
     "transformers",
     "datasets",
     "evaluate",
@@ -20,18 +19,13 @@ install_requires = [
     "streamlit-paste-button",
     "shapely",
     "pyclipper",
-    "optimum[exporters]"
-]
 
-# Add platform-specific dependencies
-if platform.system() == "Darwin":
-    install_requires.append("onnxruntime")
-else:
-    install_requires.append("onnxruntime-gpu")
+    "optimum[exporters]",
+]
 
 setup(
     name="texteller",
-    version="0.1.0",
+    version="0.1.2",
     author="OleehyO",
     author_email="1258009915@qq.com",
     description="A meta-package for installing dependencies",