add comment to divide functions/ui

src/distilabel_dataset_generator/apps/eval.py

@@ -178,6 +178,11 @@ def push_to_hub(
     new_repo_id = f"{org_name}/{repo_name}"
 
 
+######################
+# Gradio UI
+######################
+
+
 with gr.Blocks() as app:
     gr.Markdown("## 1. Select your input dataset")
     with gr.Row():

src/distilabel_dataset_generator/apps/sft.py

@@ -344,6 +344,11 @@ def push_dataset_to_argilla(
     return ""
 
 
+######################
+# Gradio UI
+######################
+
+
 with gr.Blocks(css=_LOGGED_OUT_CSS) as app:
     with gr.Column() as main_ui:
         gr.Markdown(value="## 1. Describe the dataset you want")

src/distilabel_dataset_generator/apps/textcat.py

@@ -353,6 +353,11 @@ def update_max_num_labels(labels):
     return gr.update(maximum=len(labels) if labels else 1)
 
 
+######################
+# Gradio UI
+######################
+
+
 with gr.Blocks(css=_LOGGED_OUT_CSS) as app:
     with gr.Column() as main_ui:
         gr.Markdown("## 1. Describe the dataset you want")