Upload Dockerfile with huggingface_hub

Dockerfile

@@ -1,9 +1,33 @@
+# Use the official Unsloth image which has all training deps pre-installed
 FROM unsloth/unsloth:latest
+
+USER root
 WORKDIR /app
-RUN /opt/venv/bin/python -m pip install --no-cache-dir -U huggingface_hub>=0.28.0 trl==0.15.0
-COPY ./requirements.txt requirements.txt
-RUN /opt/venv/bin/python -m pip install --no-cache-dir -r requirements.txt
-COPY . /app
-ENV PYTHONPATH=/app
+
+# Install server-specific dependencies on top of the Unsloth environment
+# We use the /opt/venv which is the standard location in Unsloth images
+RUN /opt/venv/bin/pip install --no-cache-dir \
+    fastapi \
+    uvicorn[standard] \
+    gradio \
+    openenv-core \
+    networkx \
+    matplotlib \
+    plotly \
+    pyyaml \
+    rich
+
+# Copy the project files
+COPY . .
+
+# Set environment variables
+ENV PYTHONPATH="/app"
+ENV GRADIO_ANALYTICS_ENABLED=False
+# Disable vLLM to avoid Gemma3Config import errors in the Space
 ENV UNSLOTH_USE_VLLM=0
-ENTRYPOINT ["/opt/venv/bin/python", "training/launch_hf_training.py"]
+
+# Expose the default Gradio/FastAPI port
+EXPOSE 7860
+
+# Start the FastAPI server (which mounts the Gradio demo at /demo)
+CMD ["/opt/venv/bin/python", "-m", "uvicorn", "server.main:app", "--host", "0.0.0.0", "--port", "7860"]