added download_model_comet.py code

download_model_comet.py

@@ -0,0 +1,15 @@
+from comet_ml.api import API
+from src.utils.config_loader import constants
+
+api = API()
+
+# set env variable COMET_API_KEY
+
+api.download_registry_model(
+    "anujpanthri",
+    "image-colorization-model",
+    version="1.0.0",
+    output_path=constants.ARTIFACT_MODEL_DIR,
+    expand=True,
+    stage=None, 
+)

requirements.txt

@@ -1,4 +1,77 @@
-huggingface_hub
-comet_ml
-scikit-image
-cerberus
+absl-py==2.1.0
+astunparse==1.6.3
+attrs==23.2.0
+Cerberus==1.3.5
+certifi==2024.2.2
+charset-normalizer==3.3.2
+comet-ml==3.42.1
+configobj==5.0.8
+contourpy==1.2.1
+cycler==0.12.1
+dulwich==0.22.1
+everett==3.1.0
+filelock==3.14.0
+flatbuffers==24.3.25
+fonttools==4.51.0
+fsspec==2024.5.0
+gast==0.5.4
+google-pasta==0.2.0
+grpcio==1.64.0
+h5py==3.11.0
+huggingface-hub==0.23.1
+idna==3.7
+imageio==2.34.1
+joblib==1.4.2
+jsonschema==4.22.0
+jsonschema-specifications==2023.12.1
+keras==3.3.3
+kiwisolver==1.4.5
+lazy_loader==0.4
+libclang==18.1.1
+Markdown==3.6
+markdown-it-py==3.0.0
+MarkupSafe==2.1.5
+matplotlib==3.9.0
+mdurl==0.1.2
+ml-dtypes==0.3.2
+namex==0.0.8
+networkx==3.3
+numpy==1.26.4
+opencv-python==4.9.0.80
+opt-einsum==3.3.0
+optree==0.11.0
+packaging==24.0
+pillow==10.3.0
+protobuf==4.25.3
+psutil==5.9.8
+Pygments==2.18.0
+pyparsing==3.1.2
+python-box==6.1.0
+python-dateutil==2.9.0.post0
+PyYAML==6.0.1
+referencing==0.35.1
+requests==2.32.2
+requests-toolbelt==1.0.0
+rich==13.7.1
+rpds-py==0.18.1
+scikit-image==0.23.2
+scikit-learn==1.5.0
+scipy==1.13.1
+semantic-version==2.10.0
+sentry-sdk==2.2.1
+setuptools==69.2.0
+simplejson==3.19.2
+six==1.16.0
+tensorboard==2.16.2
+tensorboard-data-server==0.7.2
+tensorflow==2.16.1
+termcolor==2.4.0
+threadpoolctl==3.5.0
+tifffile==2024.5.10
+tqdm==4.66.4
+typing_extensions==4.11.0
+urllib3==2.2.1
+Werkzeug==3.0.3
+wheel==0.43.0
+wrapt==1.16.0
+wurlitzer==3.1.0