From 569c72ffe369b68ad4a52260bebe79d0614e48ad 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: Sun, 23 Jun 2024 22:13:51 +0800
Subject: [PATCH] Remove onnxruntime-gpu

---
 setup.py | 14 ++++----------
 1 file changed, 4 insertions(+), 10 deletions(-)

diff --git a/setup.py b/setup.py
index fd37181..76f106f 100644
--- a/setup.py
+++ b/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",