alikia2x/OpenRewind
1
1
Fork 0
Code Issues Releases 1 Activity

Fix for setup.py

Browse Source
This commit is contained in:
Koen van Eijk 2024-06-18 13:32:32 +02:00
parent 932c858730
commit 3189052a41
1 changed files with 9 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
setup.py
View File

@ -11,35 +11,18 @@ install_requires = [
"Flask==3.0.3", "Flask==3.0.3",
"numpy==1.26.4", "numpy==1.26.4",
"mss==9.0.1", "mss==9.0.1",
"pillow==10.3.0",
"sentence-transformers==3.0.0", "sentence-transformers==3.0.0",
"torch==2.3.0", "torch==2.3.0",
"torchvision==0.18.0", "torchvision==0.18.0",
"shapely", "shapely==2.0.4",
"h5py", "h5py==3.11.0",
"rapidfuzz", "rapidfuzz==3.9.3",
] ]
import subprocess
import sys
def install_doctr():
subprocess.run(
[
sys.executable,
"-m",
"pip",
"install",
"git+https://github.com/koenvaneijk/doctr.git",
]
)
install_doctr()
# Define OS-specific dependencies # Define OS-specific dependencies
extras_require = {"windows": ["pywin32", "psutil"], "macos": ["pyobjc"], "linux": []} extras_require = {"windows": ["pywin32", "psutil"], "macos": ["pyobjc==10.3"], "linux": [],
'python-doctr': ['python-doctr @ git+https://github.com/koenvaneijk/doctr.git@af711bc04eb8876a7189923fb51ec44481ee18cd']
}
# Determine the current OS # Determine the current OS
current_os = platform.system().lower() current_os = platform.system().lower()
@ -56,9 +39,11 @@ else:
if current_os and current_os in extras_require: if current_os and current_os in extras_require:
install_requires.extend(extras_require[current_os]) install_requires.extend(extras_require[current_os])
install_requires.extend(extras_require.get("python-doctr", []))
setup( setup(
name="OpenRecall", name="OpenRecall",
version="0.5", version="0.6",
packages=find_packages(), packages=find_packages(),
install_requires=install_requires, install_requires=install_requires,
long_description=long_description, long_description=long_description,