Update setup.py

karimosman89 · Sep 21, 2024 · 1ba1be8 · 1ba1be8
1 parent 2a4698a
commit 1ba1be8
Showing 1 changed file with 19 additions and 15 deletions.
diff --git a/setup.py b/setup.py
@@ -1,23 +1,27 @@
 from setuptools import setup, find_packages
 
 setup(
-    name='ai_project',
-    version='0.1.0',
+    name="AI-Project-CIFAR10",
+    version="1.0",
     packages=find_packages(),
     install_requires=[
-        'flask',
-        'pandas',
-        'scikit-learn',
-        'mlflow',
-        'joblib',
+        "tensorflow==2.6.0",
+        "flask",
+        "pandas",
+        "mlflow",
+        "numpy",
+        "Pillow",
+        "airflow",
+        "boto3",
+        "kubernetes",
+        "scikit-learn",
+        "matplotlib"
     ],
     entry_points={
-        'console_scripts': [
-            'preprocess=src.preprocessing:preprocess_data',
-            'train=src.train_model:train_model',
-        ],
-    },
-    author='Your Name',
-    description='End-to-end AI project with ML training, API deployment, and CI/CD pipeline',
-    license='MIT'
+        "console_scripts": [
+            "train_model=src.train_model:train_model",
+            "preprocess_images=src.preprocessing:preprocess_images"
+        ]
+    }
 )
+