Updated FastAPI code

Dockerfile

@@ -18,7 +18,6 @@ RUN mkdir -p /tmp/huggingface /tmp/matplotlib
 RUN chown -R app:app /tmp/huggingface /tmp/matplotlib
 
 # Copy source code and set ownership
-# Do this before switching user
 COPY . .
 RUN chown -R app:app /app
 
@@ -26,8 +25,10 @@ RUN chown -R app:app /app
 USER app
 
 # Set environment variables
-# Add the user's local bin to the PATH
-ENV PATH=/home/app/.local/bin:$PATH
+# 1. Add user's local bin to the PATH for executables (like uvicorn)
+# 2. Add user's site-packages to PYTHONPATH for module imports
+ENV PATH="/home/app/.local/bin:${PATH}"
+ENV PYTHONPATH="/home/app/.local/lib/python3.10/site-packages"
 ENV HF_HOME=/tmp/huggingface
 ENV HF_HUB_CACHE=/tmp/huggingface
 ENV TRANSFORMERS_CACHE=/tmp/huggingface