Change buttons

app.py

@@ -4,6 +4,19 @@ import nbformat as nbf
 from huggingface_hub import HfApi
 
 
+"""
+TODOs:
+- Show auth and push button only after notebook creation
+- Improve the link to the result notebook
+- Handle erros
+- Add more commands to the notebook
+- Parametrize the commands
+- How to handle configs and splits?
+- Let user choose the framework
+- Improve logs
+"""
+
+
 def create_notebook_file(cell_commands, notebook_name="generated_notebook.ipynb"):
     nb = nbf.v4.new_notebook()
     nb["cells"] = [nbf.v4.new_code_cell(command) for command in cell_commands]
@@ -67,9 +80,9 @@ with gr.Blocks() as demo:
             """
         return gr.HTML(value=html_code)
 
-    generate_btn = gr.Button("Generate notebook and push to repo", visible=True)
+    generate_btn = gr.Button("Generate notebook", visible=True)
 
-    download_link = gr.File(label="Download Notebook")
+    download_link = gr.File(label="Download notebook")
     generate_btn.click(
         generate_notebook, inputs=[dataset_name], outputs=[download_link]
     )
@@ -94,7 +107,7 @@ with gr.Blocks() as demo:
             push_btn: gr.Row(visible=True),
         }
 
-    push_btn = gr.Button("Push notebook to repo", visible=False)
+    push_btn = gr.Button("Push notebook to hub", visible=False)
     token_box.change(
         auth,
         inputs=token_box,