fix: removing the nvidia libaries from requirements.txt.

fix: importing project root dir in app module.
fix: using hf_home env in docker file to store the transformers cache.

Dockerfile

@@ -18,7 +18,7 @@ libssl-dev \
 
 ENV PYTHONPATH=/app
 
-ENV TRANSFORMERS_CACHE=/tmp/cache/
+ENV HF_HOME=/tmp/cache/
 
 
 RUN pip install --no-cache-dir -r requirements.txt

app.py

@@ -1,4 +1,5 @@
 from src.insurance_assistants.ui import UI
+from src.insurance_assistants.consts import PROJECT_ROOT_DIR
 
 if __name__ == "__main__":
     UI().launch(allowed_paths=[(PROJECT_ROOT_DIR /"data/policy_wordings").as_posix()],

requirements.txt

@@ -315,40 +315,6 @@ numpy==2.2.6
     #   scipy
     #   torchvision
     #   transformers
-nvidia-cublas-cu12==12.4.5.8
-    # via
-    #   nvidia-cudnn-cu12
-    #   nvidia-cusolver-cu12
-    #   torch
-nvidia-cuda-cupti-cu12==12.4.127
-    # via torch
-nvidia-cuda-nvrtc-cu12==12.4.127
-    # via torch
-nvidia-cuda-runtime-cu12==12.4.127
-    # via torch
-nvidia-cudnn-cu12==9.1.0.70
-    # via torch
-nvidia-cufft-cu12==11.2.1.3
-    # via torch
-nvidia-curand-cu12==10.3.5.147
-    # via torch
-nvidia-cusolver-cu12==11.6.1.9
-    # via torch
-nvidia-cusparse-cu12==12.3.1.170
-    # via
-    #   nvidia-cusolver-cu12
-    #   torch
-nvidia-cusparselt-cu12==0.6.2
-    # via torch
-nvidia-nccl-cu12==2.21.5
-    # via torch
-nvidia-nvjitlink-cu12==12.4.127
-    # via
-    #   nvidia-cusolver-cu12
-    #   nvidia-cusparse-cu12
-    #   torch
-nvidia-nvtx-cu12==12.4.127
-    # via torch
 openai==1.82.1
     # via
     #   llama-index-agent-openai