kiurtis commited on
Commit
687c6ad
·
1 Parent(s): 5e62440

add installable package

Browse files
Files changed (1) hide show
  1. setup.py +44 -0
setup.py ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from setuptools import setup, find_packages
2
+
3
+ with open("README.md", "r", encoding="utf-8") as fh:
4
+ long_description = fh.read()
5
+
6
+ setup(
7
+ name="tasks",
8
+ version="0.1.0",
9
+ author="Frugal AI",
10
+ description="A package for AI tasks and utilities",
11
+ long_description=long_description,
12
+ long_description_content_type="text/markdown",
13
+ url="https://huggingface.co/spaces/Amiel/terra-cognita-frugal-ai",
14
+ packages=find_packages(),
15
+ classifiers=[
16
+ "Programming Language :: Python :: 3",
17
+ "License :: OSI Approved :: MIT License",
18
+ "Operating System :: OS Independent",
19
+ ],
20
+ python_requires=">=3.9",
21
+ install_requires=[
22
+ "fastapi>=0.68.0",
23
+ "uvicorn>=0.15.0",
24
+ "codecarbon>=2.3.1",
25
+ "datasets>=2.14.0",
26
+ "scikit-learn>=1.0.2",
27
+ "pydantic>=1.10.0",
28
+ "python-dotenv>=1.0.0",
29
+ "gradio>=4.0.0",
30
+ "requests>=2.31.0",
31
+ "librosa==0.10.2.post1",
32
+ "numpy==1.26.4",
33
+ "ultralytics==8.3.68",
34
+ "ultralytics-thop==2.0.14",
35
+ "opencv-python==4.11.0.86",
36
+ "matplotlib==3.8.1",
37
+ "onnx==1.17.0",
38
+ "tensorrt==10.8.0.43",
39
+ "tensorrt_cu12==10.8.0.43",
40
+ "tensorrt_cu12_bindings==10.8.0.43",
41
+ "tensorrt_cu12_libs==10.8.0.43",
42
+ "optimum[onnxruntime-gpu]==1.24.0"
43
+ ],
44
+ )