Modified dockerfile

Dockerfile

@@ -28,5 +28,14 @@ COPY backend/ ./backend/
 # copy built UI from the builder stage
 COPY --from=ui-builder /app/ui/out ./ui/dist
 
+ENV TRANSFORMERS_CACHE=/data/hf_cache \
+    HF_HOME=/data/hf_home \
+    HUGGINGFACE_HUB_CACHE=/data/hf_cache \
+    SENTENCE_TRANSFORMERS_HOME=/data/st_cache \
+    XDG_CACHE_HOME=/data/.cache
+
+RUN mkdir -p /data/hf_cache /data/hf_home /data/st_cache /data/.cache && \
+    chmod -R 777 /data
+
 EXPOSE 7860
 CMD ["uvicorn", "backend.api:api", "--host", "0.0.0.0", "--port", "7860"]

backend/agent.py

@@ -26,6 +26,12 @@ from langgraph.graph import StateGraph, START, END
 from langgraph.prebuilt import ToolNode
 from langgraph.checkpoint.memory import MemorySaver
 
+os.environ.setdefault("TRANSFORMERS_CACHE", "/data/hf_cache")
+os.environ.setdefault("HF_HOME", "/data/hf_home")
+os.environ.setdefault("HUGGINGFACE_HUB_CACHE", "/data/hf_cache")
+os.environ.setdefault("SENTENCE_TRANSFORMERS_HOME", "/data/st_cache")
+os.environ.setdefault("XDG_CACHE_HOME", "/data/.cache")
+
 api_key = os.environ.get("NVIDIA_API_KEY")
 if not api_key:
     raise RuntimeError("🚨 NVIDIA_API_KEY not found in environment!")