From b7da4f4a3ad97fb8c9496f614941985beb64fdf6 Mon Sep 17 00:00:00 2001
From: yan-gao-GY <yan@flower.ai>
Date: Wed, 10 Jul 2024 09:52:54 +0100
Subject: [PATCH] Add additional prompt for flowertune

---
 src/py/flwr/cli/new/new.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/py/flwr/cli/new/new.py b/src/py/flwr/cli/new/new.py
index 9367cf6c9ff..a0a2dc98556 100644
--- a/src/py/flwr/cli/new/new.py
+++ b/src/py/flwr/cli/new/new.py
@@ -264,9 +264,11 @@ def new(
             bold=True,
         )
     )
+
+    _add = "	huggingface-cli login\n" if framework_str == "flowertune" else ""
     print(
         typer.style(
-            f"	cd {package_name}\n" + "	pip install -e .\n	flwr run\n",
+            f"	cd {package_name}\n" + "	pip install -e .\n" + _add + "	flwr run\n",
             fg=typer.colors.BRIGHT_CYAN,
             bold=True,
         )