Improve comments

app.py

@@ -215,7 +215,7 @@ def generate_code(components: dict[Component, Any]):
 
     if push_to_hub:
         if not should_login:
-            create_login_hf_cells(notebook['cells'])
+            create_login_hf_cells(notebook['cells'], output_dir=output_dir)
         push_to_hub_cells(notebook['cells'], output_dir)
 
     file_name = f"{finetuning_notebook}.ipynb"

utils/components_creator.py

@@ -173,10 +173,14 @@ def add_outputs() -> (Component, Component):
                             info='The output directory where the model and checkpoints will be saved.',
                             elem_id=OUTPUT_DIR_ID)
 
-    push_to_hub = gr.Checkbox(label="push_to_hub", value=False, interactive=True,
-                              info="Whether or not to upload the trained model to the hub after training. If this is "
-                                   "True, you must specify 'HF_TOKEN'.",
-                              elem_id=PUSH_TO_HUB_ID)
+    push_to_hub = gr.Checkbox(
+        label="Push to Hub",
+        value=False,
+        interactive=True,
+        info="Select this option if you want to upload the trained model to Hugging Face Hub after training. "
+             "Please note, if this option is selected, you must provide a valid 'HF_TOKEN' in the generated notebook.",
+        elem_id=PUSH_TO_HUB_ID
+    )
 
     return output_dir, push_to_hub
 

utils/notebook_generator.py

@@ -57,22 +57,14 @@ import torch; assert torch.cuda.get_device_capability()[0] >= 8, 'Hardware not s
     cells.append(code_cell)
 
 
-def create_login_hf_cells(cells: list, should_login: bool = False, model_name: Optional[str] = None):
-    text_cell = nbf.v4.new_markdown_cell(
-        "## Login to HF")
+def create_login_hf_cells(cells: list, should_login: bool = False, model_name: Optional[str] = None,
+                          output_dir: Optional[str] = None):
+    text_cell = nbf.v4.new_markdown_cell("## Login to HF")
 
-    text_1 = """
-    Login with `HF_TOKEN` in order to push the fine-tuned model to `huggingface_hub`.
-    
-    Replace `HF_TOKEN` with a valid Token.
-    """
+    text_1 = f"Replace `HF_TOKEN` with a valid token in order to push **'{output_dir}'** to `huggingface_hub`."
 
     if should_login:
-        text_1 = f"""    
-    Login with `HF_TOKEN` in order to load **{model_name}** from `huggingface_hub`.
-
-     Replace `HF_TOKEN` with a valid Token.
-    """
+        text_1 = f"Replace `HF_TOKEN` with a valid token in order to load **'{model_name}'** from `huggingface_hub`."
 
     text_cell1 = nbf.v4.new_markdown_cell(text_1)
     code = """