Reduce image size

Dockerfile

@@ -19,7 +19,7 @@ COPY pyproject.toml uv.lock ./
 
 # Install dependencies into the virtual environment using uv
 # This is faster than pip and uses the lock file for reproducible builds.
-RUN uv sync --locked --no-cache
+RUN uv sync --locked --no-cache --index-strategy=unsafe-best-match
 
 
 # This stage creates the final, lean image for production.

flower_apps/quickstart_huggingface/pyproject.toml

@@ -12,13 +12,22 @@ authors = [
     { name = "Kaushik Amar Das", email = "kaushik.das@iiitg.ac.in" },
 ]
 dependencies = [
-    "flwr[simulation]>=1.20.0",
+    "flwr[simulation]>=1.21.0",
     "flwr-datasets>=0.5.0",
-    "torch==2.6.0",
+    "torch>=2.8.0",
     "transformers>=4.30.0,<5.0",
     "evaluate>=0.4.0,<1.0",
     "datasets>=2.0.0, <3.0",
     "scikit-learn>=1.3.1, <2.0",
+    "markupsafe==2.1.3"
+]
+
+[tool.uv.sources]
+torch = [
+    { index = "pytorch-cpu" },
+]
+torchvision = [
+    { index = "pytorch-cpu" },
 ]
 
 [tool.hatch.build.targets.wheel]

pyproject.toml

@@ -6,15 +6,28 @@ readme = "README.md"
 requires-python = ">=3.11"
 dependencies = [
     "apscheduler>=3.11.0",
-    "flwr[simulation]>=1.20.0",
+    "flwr[simulation]>=1.21.0",
     "flwr-datasets>=0.5.0",
     "gradio[oauth]>=5.44.1",
     "torch>=2.8.0",
     "transformers>=4.56.1",
     "scikit-learn>=1.7.1",
     "evaluate>=0.4.5",
+    "markupsafe==2.1.3"
 ]
 
+[tool.uv.sources]
+torch = [
+    { index = "pytorch-cpu" },
+]
+torchvision = [
+    { index = "pytorch-cpu" },
+]
+
+[[tool.uv.index]]
+name = "pytorch-cpu"
+url = "https://download.pytorch.org/whl/cpu"
+
 [tool.ruff.lint]
 
 # Enable Pyflakes (`F`) and a subset of the pycodestyle (`E`)  codes by default.