Add a loading status

app_text_classification.py

@@ -43,6 +43,10 @@ def get_demo():
         dataset_config_input = gr.Dropdown(label="Dataset Config", visible=False, allow_custom_value=True)
         dataset_split_input = gr.Dropdown(label="Dataset Split", visible=False, allow_custom_value=True)
 
+    with gr.Row():
+        first_line_ds = gr.DataFrame(label="Dataset preview", visible=False)
+    with gr.Row():
+        loading_status = gr.HTML(visible=True)
     with gr.Row():
         example_btn = gr.Button(
             "Validate model & dataset",
@@ -51,8 +55,6 @@ def get_demo():
             interactive=False,
         )
 
-    with gr.Row():
-        first_line_ds = gr.DataFrame(label="Dataset preview", visible=False)
     with gr.Row():
         example_input = gr.HTML(visible=False)
     with gr.Row():
@@ -118,19 +120,19 @@ def get_demo():
     dataset_id_input.change(
         check_dataset,
         inputs=[dataset_id_input],
-        outputs=[dataset_config_input, dataset_split_input, first_line_ds],
+        outputs=[dataset_config_input, dataset_split_input, first_line_ds, loading_status],
     )
 
     dataset_config_input.change(
         check_dataset,
         inputs=[dataset_id_input, dataset_config_input],
-        outputs=[dataset_config_input, dataset_split_input, first_line_ds],
+        outputs=[dataset_config_input, dataset_split_input, first_line_ds, loading_status],
     )
 
     dataset_split_input.change(
         check_dataset,
         inputs=[dataset_id_input, dataset_config_input, dataset_split_input],
-        outputs=[dataset_config_input, dataset_split_input, first_line_ds],
+        outputs=[dataset_config_input, dataset_split_input, first_line_ds, loading_status],
     )
 
     scanners.change(write_scanners, inputs=[scanners, uid_label])
@@ -180,7 +182,7 @@ def get_demo():
             dataset_config_input,
             dataset_split_input,
         ],
-        outputs=[example_btn],
+        outputs=[example_btn, loading_status],
     )
 
     gr.on(
@@ -200,6 +202,7 @@ def get_demo():
             example_prediction,
             column_mapping_accordion,
             run_btn,
+            loading_status,
             *column_mappings,
         ],
     )

text_classification_ui_helpers.py

@@ -73,12 +73,14 @@ def check_dataset(dataset_id, dataset_config=None, dataset_split=None):
                 gr.Dropdown(configs, value=configs[0], visible=True),
                 gr.Dropdown(splits, value=splits[0], visible=True),
                 gr.DataFrame(pd.DataFrame(), visible=False),
+                "",
             )
         elif dataset_split == None:
             return (
                 gr.Dropdown(configs, value=dataset_config, visible=True),
                 gr.Dropdown(splits, value=splits[0], visible=True),
                 gr.DataFrame(pd.DataFrame(), visible=False),
+                "",
             )
 
     dataset_dict = datasets.load_dataset(dataset_id, dataset_config)
@@ -87,6 +89,7 @@ def check_dataset(dataset_id, dataset_config=None, dataset_split=None):
         gr.Dropdown(configs, value=dataset_config, visible=True),
         gr.Dropdown(splits, value=dataset_split, visible=True),
         gr.DataFrame(dataframe, visible=True),
+        "",
     )
 
 
@@ -192,15 +195,15 @@ def precheck_model_ds_enable_example_btn(
     ppl = check_model(model_id)
     if ppl is None or not isinstance(ppl, TextClassificationPipeline):
         gr.Warning("Please check your model.")
-        return gr.update(interactive=False)
+        return gr.update(interactive=False), ""
     ds_labels, ds_features = get_labels_and_features_from_dataset(
         dataset_id, dataset_config, dataset_split
     )
     if not isinstance(ds_labels, list) or not isinstance(ds_features, list):
         gr.Warning(CHECK_CONFIG_OR_SPLIT_RAW)
-        return gr.update(interactive=False)
+        return gr.update(interactive=False), ""
 
-    return gr.update(interactive=True)
+    return gr.update(interactive=True), ""
 
 
 def align_columns_and_show_prediction(
@@ -214,6 +217,7 @@ def align_columns_and_show_prediction(
             gr.update(visible=False),
             gr.update(visible=False, open=False),
             gr.update(interactive=False),
+            "",
             *[gr.update(visible=False) for _ in range(MAX_LABELS + MAX_FEATURES)],
         )
 
@@ -243,6 +247,7 @@ def align_columns_and_show_prediction(
             gr.update(visible=False),
             gr.update(visible=False, open=False),
             gr.update(interactive=False),
+            "",
             *dropdown_placement,
         )
 
@@ -264,6 +269,7 @@ def align_columns_and_show_prediction(
             gr.update(visible=False),
             gr.update(visible=True, open=True),
             gr.update(interactive=True),
+            "",
             *column_mappings,
         )
 
@@ -275,6 +281,7 @@ def align_columns_and_show_prediction(
         gr.update(value=prediction_output, visible=True),
         gr.update(visible=True, open=False),
         gr.update(interactive=True),
+        "",
         *column_mappings,
     )