Spaces:

nithin04
/

graphforge-openenv

Sleeping

NagaNithin-V

Switch to Gradio demo — load trained LoRA or base model

ffe45d8 6 months ago

797 Bytes

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"]

	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"]