changed the docker

Dockerfile

@@ -1,73 +1,41 @@
-# Use NVIDIA CUDA base image (Ubuntu 22.04 jammy)
+# Use NVIDIA CUDA base image (Ubuntu 22.04)
 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
-
-# ---------------- System deps + Python 3.12 + headless GL ----------------
+# System deps (Python 3.12 + basics)
 RUN apt-get update && apt-get install -y --no-install-recommends \
-    software-properties-common \
-    ca-certificates \
-    curl wget git build-essential \
+    python3.12 python3.12-venv python3-pip python3.12-distutils \
+    git curl ca-certificates build-essential \
     libgl1-mesa-glx libglib2.0-0 libsm6 libxext6 libxrender1 \
-    libgomp1 libgcc-s1 \
     libosmesa6 libosmesa6-dev \
     libegl1-mesa libegl1-mesa-dev \
- && add-apt-repository ppa:deadsnakes/ppa -y \
- && apt-get update && apt-get install -y --no-install-recommends \
-    python3.12 python3.12-dev python3.12-venv \
  && rm -rf /var/lib/apt/lists/*
 
-# Install pip for Python 3.12
-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
-RUN ln -sf /usr/bin/python3.12 /usr/bin/python && \
-    ln -sf /usr/local/bin/pip3.12 /usr/local/bin/pip
-
-# ---------------- Writable app data & HF cache ----------------
-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
+# Optional: make `python` and `pip` point to 3.12
+RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.12 1 && \
+    update-alternatives --install /usr/bin/pip pip /usr/bin/pip3 1
 
-RUN mkdir -p /data/geometry /data/solution /data/weights /data/hf_home /data/matplotlib \
- && chmod -R 777 /data
-
-# ---------------- Install frpc for Gradio share=True ----------------
-RUN mkdir -p /data/hf_home/gradio/frpc && \
-    wget https://cdn-media.huggingface.co/frpc-gradio-0.3/frpc_linux_amd64 -O /data/hf_home/gradio/frpc/frpc_linux_amd64_v0.3 && \
-    chmod +x /data/hf_home/gradio/frpc/frpc_linux_amd64_v0.3
-
-# ---------------- Application setup ----------------
+# Workdir
 WORKDIR /app
 
-# Copy requirements first (better caching)
+# Copy requirements first (good for build caching)
 COPY requirements.txt .
 
-# Install Python deps
-RUN pip install --no-cache-dir --upgrade pip setuptools wheel \
- && pip install --no-cache-dir -r requirements.txt
+# Install Python packages
+RUN pip install --upgrade pip && \
+    pip install --no-cache-dir -r requirements.txt
 
-# Copy the rest of the source code
+# Copy the rest of your app
 COPY . .
 
-# Permissions
-RUN chmod +x app.py
-RUN chown -R 1000:1000 /app
-
-# Expose Trame/Gradio port
+# HF Spaces default port
 EXPOSE 7860
 
-# Run app (Trame or Gradio, same entrypoint)
-CMD ["python", "app.py"]
+# Start your Trame app (change app_trame.py to your actual file)
+CMD ["python", "app_trame.py"]

requirements.txt

@@ -8,9 +8,9 @@
 torch==2.5.0
 torchvision==0.20.0
 torchaudio==2.5.0
-torch_geometric
 
 -f https://data.pyg.org/whl/torch-2.5.0+cu121.html
+torch_geometric
 pyg_lib
 torch_scatter
 torch_sparse