IndicPhotoOCR / setup.py
anikde's picture
initial commit
e2f99d5
from setuptools import setup, find_packages
setup(
name="IndicPhotoOCR",
version="1.1.0",
description="Scene Text Recognition Toolkit across 13 Indian Languages which contains detection, script identification, and text recognition modules",
long_description=open("README.md").read() + "\n\n" + open("CHANGELOG.md").read(),
long_description_content_type="text/markdown",
author="Anik De",
author_email="[email protected]",
url="https://github.com/Bhashini-IITJ/IndicPhotoOCR",
packages=find_packages(),
python_requires='>=3.9',
install_requires=[
# Your mandatory dependencies here
'aiohappyeyeballs==2.4.3',
'aiohttp==3.10.10',
'aiosignal==1.3.1',
'async-timeout==4.0.3',
'attrs==24.2.0',
'certifi==2024.8.30',
'charset-normalizer==3.4.0',
'click==8.1.7',
'filelock==3.16.1',
'frozenlist==1.5.0',
'fsspec==2024.10.0',
'huggingface-hub==0.26.1',
'idna==3.10',
'jinja2==3.1.4',
'joblib==1.4.2',
'lightning-utilities==0.11.8',
'markupsafe==3.0.2',
'mpmath==1.3.0',
'multidict==6.1.0',
'networkx==3.2.1',
'nltk==3.9.1',
'numpy==1.26.4',
'packaging==24.1',
'pillow==11.0.0',
'propcache==0.2.0',
'pytorch-lightning==2.4.0',
'pyyaml==6.0.2',
'regex==2024.9.11',
'requests==2.32.3',
'safetensors==0.4.5',
'sympy==1.13.1',
'timm==1.0.11',
'torchmetrics==1.5.1',
'tqdm==4.66.5',
'typing-extensions==4.12.2',
'urllib3==2.2.3',
'yarl==1.16.0',
'opencv-python==4.10.0.84',
'shapely==2.0.6',
'openai-clip==1.0.1',
'lmdb==1.5.1'
],
extras_require={
'cu118': [
'torch==2.5.0+cu118',
'torchvision==0.20.0+cu118',
# Any additional packages specific to cu118
],
'cu121': [
'torch==2.5.0+cu121',
'torchvision==0.20.0+cu121',
# Any additional packages specific to cu121
],
'cpu': [
'torch==2.5.0',
'torchvision==0.20.0',
# Any additional packages specific to CPU
],
'extra': [
'six==1.16.0', # Your other extra requirements
],
},
)