Switch to Gradio demo — load trained LoRA or base model

Dockerfile

@@ -1,27 +1,32 @@
-# Hugging Face Space Dockerfile.
-# Mirrors the root Dockerfile, exists separately because HF Spaces looks for
-# the Dockerfile inside the Space root by default.
-
 FROM python:3.11-slim
 
 WORKDIR /app
 
-COPY pyproject.toml ./
-COPY graphforge ./graphforge
-COPY env ./env
-COPY openenv.yaml ./
+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" \
-    "fastapi>=0.110" \
-    "uvicorn[standard]>=0.27" \
-    "httpx>=0.27" \
-    "openenv-core>=0.1.0" \
-    "pyyaml>=6.0"
+    "pyyaml>=6.0" \
+    gitpython
+
+# Install project
+RUN pip install --no-cache-dir -e "." || true
 
 ENV PYTHONUNBUFFERED=1
 ENV PYTHONPATH=/app
 
 EXPOSE 7860
 
-CMD ["uvicorn", "env.server:app", "--host", "0.0.0.0", "--port", "7860"]
+CMD ["python", "training/app.py"]