Ajitg25 commited on
Commit
56caa44
·
verified ·
1 Parent(s): 4d18b99

Fix: CPU-only Dockerfile running showcase.py (no model/training)

Browse files
Files changed (1) hide show
  1. Dockerfile +19 -7
Dockerfile CHANGED
@@ -1,20 +1,32 @@
1
- FROM pytorch/pytorch:2.4.0-cuda12.4-cudnn9-runtime
2
 
3
- RUN pip install --no-cache-dir transformers>=4.45.0 accelerate && \
4
- pip install --no-cache-dir nest_asyncio peft trl \
5
- pydantic fastapi uvicorn websockets httpx matplotlib
6
 
7
- RUN pip install --no-cache-dir "openenv-core[core]>=0.2.2"
 
 
8
 
9
  RUN useradd -m -u 1000 appuser
 
10
  WORKDIR /app
 
 
 
 
11
  COPY . .
12
- RUN mkdir -p /app/plots && chown -R appuser:appuser /app
 
 
 
13
  USER appuser
14
 
 
15
  ENV PYTHONPATH="/app/envs:/app:$PYTHONPATH"
16
  ENV AMBULANCE_DIFFICULTY="easy"
17
 
18
  EXPOSE 7860
19
 
20
- CMD ["sh", "-c", "python3 /app/train.py; echo 'Training done (exit $?). Serving plots...'; mkdir -p /app/plots && cd /app/plots && python3 -m http.server 7860"]
 
 
1
+ FROM python:3.12-slim
2
 
3
+ RUN apt-get update && \
4
+ apt-get install -y --no-install-recommends git curl && \
5
+ rm -rf /var/lib/apt/lists/*
6
 
7
+ RUN curl -LsSf https://astral.sh/uv/install.sh | sh && \
8
+ mv /root/.local/bin/uv /usr/local/bin/uv && \
9
+ mv /root/.local/bin/uvx /usr/local/bin/uvx
10
 
11
  RUN useradd -m -u 1000 appuser
12
+
13
  WORKDIR /app
14
+
15
+ COPY pyproject.toml uv.lock* ./
16
+ RUN uv sync --no-install-project --no-editable || true
17
+
18
  COPY . .
19
+ RUN uv sync --no-editable
20
+
21
+ RUN chown -R appuser:appuser /app
22
+
23
  USER appuser
24
 
25
+ ENV PATH="/app/.venv/bin:$PATH"
26
  ENV PYTHONPATH="/app/envs:/app:$PYTHONPATH"
27
  ENV AMBULANCE_DIFFICULTY="easy"
28
 
29
  EXPOSE 7860
30
 
31
+ # Showcase server: serves frontend + env API on one port
32
+ CMD ["python", "showcase.py"]