added trame reqs

Dockerfile

@@ -1,20 +1,18 @@
+
 # Use NVIDIA CUDA base image (Ubuntu 22.04 jammy)
 FROM nvidia/cuda:12.3.2-devel-ubuntu22.04
 
-# ---------------- Base env ----------------
+# Non-interactive apt
 ENV DEBIAN_FRONTEND=noninteractive
 ENV PYTHONUNBUFFERED=1
 ENV CUDA_HOME=/usr/local/cuda
 ENV PATH=${CUDA_HOME}/bin:/usr/local/bin:${PATH}
 ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
 
-# Fix OpenMP threads to a valid integer (libgomp warning)
 ENV OMP_NUM_THREADS=4
 
-# Force headless: never try X11 DISPLAY
-# ENV DISPLAY=""
 
-# ---------------- System deps + Python 3.12 + headless GL ----------------
+# System deps + Deadsnakes PPA for Python 3.12
 RUN apt-get update && apt-get install -y --no-install-recommends \
     software-properties-common \
     ca-certificates \
@@ -33,7 +31,7 @@ RUN curl -sS https://bootstrap.pypa.io/get-pip.py -o /tmp/get-pip.py \
  && python3.12 /tmp/get-pip.py \
  && rm /tmp/get-pip.py
 
-# Make 'python' and 'pip' point to Python 3.12
+# Make 'python' and 'pip' point to 3.12
 RUN ln -sf /usr/bin/python3.12 /usr/bin/python && \
     ln -sf /usr/local/bin/pip3.12 /usr/local/bin/pip
 
@@ -42,7 +40,7 @@ ENV APP_DATA_DIR=/data
 ENV HF_HOME=/data/hf_home
 ENV HUGGINGFACE_HUB_CACHE=/data/hf_home
 ENV TRANSFORMERS_CACHE=/data/hf_home
-ENV MPLCONFIGDIR=/data/matplotlib
+ENV MPLCONFIGDIR=/data/matplotlib   
 
 RUN mkdir -p /data/geometry /data/solution /data/weights /data/hf_home /data/matplotlib \
  && chmod -R 777 /data
@@ -62,17 +60,19 @@ COPY requirements.txt .
 RUN pip install --no-cache-dir --upgrade pip setuptools wheel \
  && pip install --no-cache-dir -r requirements.txt
 
-# Copy the rest of the source code
+# Copy the rest of the source
 COPY . .
 
+# Create cache directory for downloaded repos
+RUN mkdir -p /app/cache && chmod 777 /app/cache
+
 # Permissions
 RUN chmod +x app.py
 RUN chown -R 1000:1000 /app
 
-# Expose Trame/Gradio port
+# Expose Gradio port
 EXPOSE 7860
 
-# Run app (Trame or Gradio, same entrypoint)
-# CMD ["python", "app.py"]
-# Run app on 0.0.0.0 so HF proxy can reach it
-CMD python app.py --host 0.0.0.0 --port ${PORT:-7860} --server
+# Run app
+CMD ["python", "app.py"]
+