Spaces:
Sleeping
Sleeping
File size: 797 Bytes
7952f32 ffe45d8 7952f32 ffe45d8 7952f32 ffe45d8 7952f32 ffe45d8 7952f32 ffe45d8 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 | FROM python:3.11-slim
WORKDIR /app
RUN apt-get update -qq && apt-get install -y -qq git && rm -rf /var/lib/apt/lists/*
# Clone repo (includes env, graphforge, training, pyproject.toml)
RUN git clone --depth 1 https://github.com/nithin062006/scaler.git .
# CPU-only torch (keeps image small; no GPU needed for 0.5B demo)
RUN pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cpu
# ML stack + Gradio
RUN pip install --no-cache-dir \
"transformers>=4.40" \
"peft>=0.10" \
"huggingface_hub>=0.22" \
"gradio>=4.36" \
"datasets>=2.18" \
"pydantic>=2.6" \
"pyyaml>=6.0" \
gitpython
# Install project
RUN pip install --no-cache-dir -e "." || true
ENV PYTHONUNBUFFERED=1
ENV PYTHONPATH=/app
EXPOSE 7860
CMD ["python", "training/app.py"]
|