added basic files and updated env

.python-version

@@ -1 +1 @@
-3.11
+3.11.10

FER/prediction.py

@@ -1,5 +1,9 @@
-from main import *
+from models.PosterV2_7cls import pyramid_trans_expr2
+import torch
+import os
 from deepface import DeepFace
+from torchvision import transforms
+import numpy as np
 
 # Checking for all types of devices available
 if torch.backends.mps.is_available():
@@ -95,7 +99,7 @@ def predict(model, image_path):
 
         print(
             f"-->Image Path {image_path} [!] The predicted labels are {y_pred} and the label is {labels}"
-            )
+        )
     return
 
 

pyproject.toml

@@ -5,24 +5,129 @@ description = "Add your description here"
 readme = "README.md"
 requires-python = ">=3.11"
 dependencies = [
-    "huggingface-hub>=0.26.2",
+    "absl-py==2.1.0",
+    "accelerate==1.1.1",
+    "aiohappyeyeballs==2.4.3",
+    "aiohttp==3.10.10",
+    "aiosignal==1.3.1",
+    "astunparse==1.6.3",
+    "attrs==24.2.0",
+    "beautifulsoup4==4.12.3",
+    "bitsandbytes==0.44.1",
+    "blinker==1.9.0",
+    "certifi==2024.8.30",
+    "charset-normalizer==3.4.0",
+    "click==8.1.7",
+    "colorama==0.4.6",
+    "colorama==0.4.6 ; platform_system == 'Windows'",
+    "contourpy==1.3.0",
+    "cycler==0.12.1",
+    "datasets==3.1.0",
+    "deepface==0.0.93",
+    "dill==0.3.8",
+    "filelock==3.16.1",
+    "fire==0.7.0",
+    "flask-cors==5.0.0",
+    "flask==3.0.3",
+    "flatbuffers==24.3.25",
+    "fonttools==4.54.1",
+    "frozenlist==1.5.0",
+    "fsspec==2024.9.0",
+    "gast==0.6.0",
+    "gdown==5.2.0",
+    "google-pasta==0.2.0",
+    "grpcio==1.67.1",
+    "gunicorn==23.0.0",
+    "h5py==3.12.1",
+    "huggingface-hub==0.26.2",
+    "idna==3.10",
+    "itsdangerous==2.2.0",
+    "jinja2==3.1.4",
+    "joblib==1.4.2",
+    "jsonlines==4.0.0",
+    "keras==3.6.0",
+    "kiwisolver==1.4.7",
+    "libclang==18.1.1",
+    "lz4==4.3.3",
+    "markdown-it-py==3.0.0",
+    "markdown==3.7",
+    "markupsafe==3.0.1",
+    "matplotlib==3.9.2",
+    "mdurl==0.1.2",
+    "ml-dtypes==0.4.1",
+    "mpmath==1.3.0",
+    "mtcnn==1.0.0",
+    "multidict==6.1.0",
+    "multiprocess==0.70.16",
+    "namex==0.0.8",
+    "networkx==3.4.2",
+    "numpy==2.0.2",
+    "nvidia-cublas-cu12==12.4.5.8 ; platform_machine == 'x86_64' and platform_system == 'Linux'",
+    "nvidia-cuda-cupti-cu12==12.4.127 ; platform_machine == 'x86_64' and platform_system == 'Linux'",
+    "nvidia-cuda-nvrtc-cu12==12.4.127 ; platform_machine == 'x86_64' and platform_system == 'Linux'",
+    "nvidia-cuda-runtime-cu12==12.4.127 ; platform_machine == 'x86_64' and platform_system == 'Linux'",
+    "nvidia-cudnn-cu12==9.1.0.70 ; platform_machine == 'x86_64' and platform_system == 'Linux'",
+    "nvidia-cufft-cu12==11.2.1.3 ; platform_machine == 'x86_64' and platform_system == 'Linux'",
+    "nvidia-curand-cu12==10.3.5.147 ; platform_machine == 'x86_64' and platform_system == 'Linux'",
+    "nvidia-cusolver-cu12==11.6.1.9 ; platform_machine == 'x86_64' and platform_system == 'Linux'",
+    "nvidia-cusparse-cu12==12.3.1.170 ; platform_machine == 'x86_64' and platform_system == 'Linux'",
+    "nvidia-nccl-cu12==2.21.5 ; platform_machine == 'x86_64' and platform_system == 'Linux'",
+    "nvidia-nvjitlink-cu12==12.4.127 ; platform_machine == 'x86_64' and platform_system == 'Linux'",
+    "nvidia-nvtx-cu12==12.4.127 ; platform_machine == 'x86_64' and platform_system == 'Linux'",
+    "opencv-python==4.10.0.84",
+    "opt-einsum==3.4.0",
+    "optree==0.13.0",
+    "packaging==24.2",
+    "pandas==2.2.3",
+    "peft==0.13.2",
+    "pillow==11.0.0",
+    "propcache==0.2.0",
+    "protobuf==5.28.3",
+    "psutil==6.1.0",
+    "pyarrow==18.0.0",
+    "pygments==2.18.0",
+    "pyparsing==3.2.0",
+    "pysocks==1.7.1",
+    "python-dateutil==2.9.0.post0",
+    "pytz==2024.2",
+    "pyyaml==6.0.2",
+    "regex==2024.11.6",
+    "requests==2.32.3",
+    "retina-face==0.0.17",
+    "rich==13.9.4",
+    "safetensors==0.4.5",
+    "scikit-learn==1.5.2",
+    "scipy==1.14.1",
+    "setuptools==75.4.0",
+    "six==1.16.0",
+    "soupsieve==2.6",
+    "sympy==1.13.1",
+    "tensorboard-data-server==0.7.2",
+    "tensorboard==2.18.0",
+    "tensorflow-intel==2.18.0",
+    "tensorflow-io-gcs-filesystem==0.31.0",
+    "tensorflow-io-gcs-filesystem==0.31.0 ; python_full_version < '3.12'",
+    "tensorflow==2.18.0",
+    "termcolor==2.5.0",
+    "threadpoolctl==3.5.0",
+    "tokenizers==0.20.3",
     "torch==2.5.1+cu124",
     "torchaudio==2.5.1+cu124",
+    "torchsampler==0.1.2",
     "torchvision==0.20.1+cu124",
+    "tqdm==4.67.0",
+    "transformers==4.46.2",
+    "triton==3.1.0 ; python_full_version < '3.13' and platform_machine == 'x86_64' and platform_system == 'Linux'",
+    "trl==0.12.0",
+    "typing-extensions==4.12.2",
+    "tzdata==2024.2",
+    "urllib3==2.2.3",
+    "werkzeug==3.1.3",
+    "wheel==0.45.0",
+    "wrapt==1.16.0",
+    "xxhash==3.5.0",
+    "yarl==1.17.1",
 ]
 
-[tool.uv.sources]
-torch = {index = "pytorch"}
-torchaudio = {index = "pytorch"}
-torchvision = {index = "pytorch"}
-cudf-cu12 = {index = "pandas_cuda"}
-
 [[tool.uv.index]]
-name = "pytorch"
 url = "https://download.pytorch.org/whl/cu124"
-explicit = true
-
-[[tool.uv.index]]
-name = "pandas_cuda"
-url = "https://pypi.nvidia.com"
-explicit = true

requirements.in

@@ -0,0 +1,18 @@
+-i https://pypi.org/simple
+--extra-index-url https://download.pytorch.org/whl/cu124
+huggingface-hub==0.26.2
+scikit-learn==1.5.2
+torch==2.5.1+cu124
+torchaudio==2.5.1+cu124
+torchsampler==0.1.2
+torchvision==0.20.1+cu124
+matplotlib==3.9.2
+deepface==0.0.93
+transformers==4.46.2
+datasets==3.1.0
+peft==0.13.2
+trl==0.12.0
+bitsandbytes==0.44.1
+accelerate==1.1.1
+tensorboard==2.18.0
+jsonlines==4.0.0

requirements.txt

@@ -0,0 +1,402 @@
+# This file was autogenerated by uv via the following command:
+#    uv pip compile requirements.in -o requirements.txt --python-version 3.11.10 --index-strategy unsafe-best-match --universal
+absl-py==2.1.0
+    # via
+    #   keras
+    #   tensorboard
+    #   tensorflow
+accelerate==1.1.1
+    # via
+    #   -r requirements.in
+    #   peft
+    #   trl
+aiohappyeyeballs==2.4.3
+    # via aiohttp
+aiohttp==3.10.10
+    # via
+    #   datasets
+    #   fsspec
+aiosignal==1.3.1
+    # via aiohttp
+astunparse==1.6.3
+    # via tensorflow
+attrs==24.2.0
+    # via
+    #   aiohttp
+    #   jsonlines
+beautifulsoup4==4.12.3
+    # via gdown
+bitsandbytes==0.44.1
+    # via -r requirements.in
+blinker==1.9.0
+    # via flask
+certifi==2024.8.30
+    # via requests
+charset-normalizer==3.4.0
+    # via requests
+click==8.1.7
+    # via flask
+colorama==0.4.6 ; platform_system == 'Windows'
+    # via
+    #   click
+    #   tqdm
+contourpy==1.3.0
+    # via matplotlib
+cycler==0.12.1
+    # via matplotlib
+datasets==3.1.0
+    # via
+    #   -r requirements.in
+    #   trl
+deepface==0.0.93
+    # via -r requirements.in
+dill==0.3.8
+    # via
+    #   datasets
+    #   multiprocess
+filelock==3.16.1
+    # via
+    #   datasets
+    #   gdown
+    #   huggingface-hub
+    #   torch
+    #   transformers
+    #   triton
+fire==0.7.0
+    # via deepface
+flask==3.0.3
+    # via
+    #   deepface
+    #   flask-cors
+flask-cors==5.0.0
+    # via deepface
+flatbuffers==24.3.25
+    # via tensorflow
+fonttools==4.54.1
+    # via matplotlib
+frozenlist==1.5.0
+    # via
+    #   aiohttp
+    #   aiosignal
+fsspec==2024.9.0
+    # via
+    #   datasets
+    #   huggingface-hub
+    #   torch
+gast==0.6.0
+    # via tensorflow
+gdown==5.2.0
+    # via
+    #   deepface
+    #   retina-face
+google-pasta==0.2.0
+    # via tensorflow
+grpcio==1.67.1
+    # via
+    #   tensorboard
+    #   tensorflow
+gunicorn==23.0.0
+    # via deepface
+h5py==3.12.1
+    # via
+    #   keras
+    #   tensorflow
+huggingface-hub==0.26.2
+    # via
+    #   -r requirements.in
+    #   accelerate
+    #   datasets
+    #   peft
+    #   tokenizers
+    #   transformers
+idna==3.10
+    # via
+    #   requests
+    #   yarl
+itsdangerous==2.2.0
+    # via flask
+jinja2==3.1.4
+    # via
+    #   flask
+    #   torch
+joblib==1.4.2
+    # via
+    #   mtcnn
+    #   scikit-learn
+jsonlines==4.0.0
+    # via -r requirements.in
+keras==3.6.0
+    # via
+    #   deepface
+    #   tensorflow
+kiwisolver==1.4.7
+    # via matplotlib
+libclang==18.1.1
+    # via tensorflow
+lz4==4.3.3
+    # via mtcnn
+markdown==3.7
+    # via tensorboard
+markdown-it-py==3.0.0
+    # via rich
+markupsafe==3.0.1
+    # via
+    #   jinja2
+    #   werkzeug
+matplotlib==3.9.2
+    # via -r requirements.in
+mdurl==0.1.2
+    # via markdown-it-py
+ml-dtypes==0.4.1
+    # via
+    #   keras
+    #   tensorflow
+mpmath==1.3.0
+    # via sympy
+mtcnn==1.0.0
+    # via deepface
+multidict==6.1.0
+    # via
+    #   aiohttp
+    #   yarl
+multiprocess==0.70.16
+    # via datasets
+namex==0.0.8
+    # via keras
+networkx==3.4.2
+    # via torch
+numpy==2.0.2
+    # via
+    #   accelerate
+    #   bitsandbytes
+    #   contourpy
+    #   datasets
+    #   deepface
+    #   h5py
+    #   keras
+    #   matplotlib
+    #   ml-dtypes
+    #   opencv-python
+    #   pandas
+    #   peft
+    #   retina-face
+    #   scikit-learn
+    #   scipy
+    #   tensorboard
+    #   tensorflow
+    #   torchvision
+    #   transformers
+nvidia-cublas-cu12==12.4.5.8 ; platform_machine == 'x86_64' and platform_system == 'Linux'
+    # via
+    #   nvidia-cudnn-cu12
+    #   nvidia-cusolver-cu12
+    #   torch
+nvidia-cuda-cupti-cu12==12.4.127 ; platform_machine == 'x86_64' and platform_system == 'Linux'
+    # via torch
+nvidia-cuda-nvrtc-cu12==12.4.127 ; platform_machine == 'x86_64' and platform_system == 'Linux'
+    # via torch
+nvidia-cuda-runtime-cu12==12.4.127 ; platform_machine == 'x86_64' and platform_system == 'Linux'
+    # via torch
+nvidia-cudnn-cu12==9.1.0.70 ; platform_machine == 'x86_64' and platform_system == 'Linux'
+    # via torch
+nvidia-cufft-cu12==11.2.1.3 ; platform_machine == 'x86_64' and platform_system == 'Linux'
+    # via torch
+nvidia-curand-cu12==10.3.5.147 ; platform_machine == 'x86_64' and platform_system == 'Linux'
+    # via torch
+nvidia-cusolver-cu12==11.6.1.9 ; platform_machine == 'x86_64' and platform_system == 'Linux'
+    # via torch
+nvidia-cusparse-cu12==12.3.1.170 ; platform_machine == 'x86_64' and platform_system == 'Linux'
+    # via
+    #   nvidia-cusolver-cu12
+    #   torch
+nvidia-nccl-cu12==2.21.5 ; platform_machine == 'x86_64' and platform_system == 'Linux'
+    # via torch
+nvidia-nvjitlink-cu12==12.4.127 ; platform_machine == 'x86_64' and platform_system == 'Linux'
+    # via
+    #   nvidia-cufft-cu12
+    #   nvidia-cusolver-cu12
+    #   nvidia-cusparse-cu12
+    #   torch
+nvidia-nvtx-cu12==12.4.127 ; platform_machine == 'x86_64' and platform_system == 'Linux'
+    # via torch
+opencv-python==4.10.0.84
+    # via
+    #   deepface
+    #   retina-face
+opt-einsum==3.4.0
+    # via tensorflow
+optree==0.13.0
+    # via keras
+packaging==24.2
+    # via
+    #   accelerate
+    #   datasets
+    #   gunicorn
+    #   huggingface-hub
+    #   keras
+    #   matplotlib
+    #   peft
+    #   tensorboard
+    #   tensorflow
+    #   transformers
+pandas==2.2.3
+    # via
+    #   datasets
+    #   deepface
+    #   torchsampler
+peft==0.13.2
+    # via -r requirements.in
+pillow==11.0.0
+    # via
+    #   deepface
+    #   matplotlib
+    #   retina-face
+    #   torchvision
+propcache==0.2.0
+    # via yarl
+protobuf==5.28.3
+    # via
+    #   tensorboard
+    #   tensorflow
+psutil==6.1.0
+    # via
+    #   accelerate
+    #   peft
+pyarrow==18.0.0
+    # via datasets
+pygments==2.18.0
+    # via rich
+pyparsing==3.2.0
+    # via matplotlib
+pysocks==1.7.1
+    # via requests
+python-dateutil==2.9.0.post0
+    # via
+    #   matplotlib
+    #   pandas
+pytz==2024.2
+    # via pandas
+pyyaml==6.0.2
+    # via
+    #   accelerate
+    #   datasets
+    #   huggingface-hub
+    #   peft
+    #   transformers
+regex==2024.11.6
+    # via transformers
+requests==2.32.3
+    # via
+    #   datasets
+    #   deepface
+    #   gdown
+    #   huggingface-hub
+    #   tensorflow
+    #   transformers
+retina-face==0.0.17
+    # via deepface
+rich==13.9.4
+    # via
+    #   keras
+    #   trl
+safetensors==0.4.5
+    # via
+    #   accelerate
+    #   peft
+    #   transformers
+scikit-learn==1.5.2
+    # via -r requirements.in
+scipy==1.14.1
+    # via scikit-learn
+setuptools==75.4.0
+    # via
+    #   tensorboard
+    #   tensorflow
+    #   torch
+six==1.16.0
+    # via
+    #   astunparse
+    #   google-pasta
+    #   python-dateutil
+    #   tensorboard
+    #   tensorflow
+soupsieve==2.6
+    # via beautifulsoup4
+sympy==1.13.1
+    # via torch
+tensorboard==2.18.0
+    # via
+    #   -r requirements.in
+    #   tensorflow
+tensorboard-data-server==0.7.2
+    # via tensorboard
+tensorflow==2.18.0
+    # via
+    #   deepface
+    #   retina-face
+tensorflow-io-gcs-filesystem==0.31.0 ; python_full_version < '3.12'
+    # via tensorflow
+termcolor==2.5.0
+    # via
+    #   fire
+    #   tensorflow
+threadpoolctl==3.5.0
+    # via scikit-learn
+tokenizers==0.20.3
+    # via transformers
+torch==2.5.1+cu124
+    # via
+    #   -r requirements.in
+    #   accelerate
+    #   bitsandbytes
+    #   peft
+    #   torchaudio
+    #   torchsampler
+    #   torchvision
+torchaudio==2.5.1+cu124
+    # via -r requirements.in
+torchsampler==0.1.2
+    # via -r requirements.in
+torchvision==0.20.1+cu124
+    # via
+    #   -r requirements.in
+    #   torchsampler
+tqdm==4.67.0
+    # via
+    #   datasets
+    #   deepface
+    #   gdown
+    #   huggingface-hub
+    #   peft
+    #   transformers
+transformers==4.46.2
+    # via
+    #   -r requirements.in
+    #   peft
+    #   trl
+triton==3.1.0 ; python_full_version < '3.13' and platform_machine == 'x86_64' and platform_system == 'Linux'
+    # via torch
+trl==0.12.0
+    # via -r requirements.in
+typing-extensions==4.12.2
+    # via
+    #   huggingface-hub
+    #   optree
+    #   tensorflow
+    #   torch
+tzdata==2024.2
+    # via pandas
+urllib3==2.2.3
+    # via requests
+werkzeug==3.1.3
+    # via
+    #   flask
+    #   tensorboard
+wheel==0.45.0
+    # via astunparse
+wrapt==1.16.0
+    # via tensorflow
+xxhash==3.5.0
+    # via datasets
+yarl==1.17.1
+    # via aiohttp