fix-prediction-input-can-be-too-long

app_text_classification.py

@@ -72,6 +72,7 @@ def get_demo():
     with gr.Row():
         validation_result = gr.HTML(visible=False)
     with gr.Row():
+        example_input = gr.Textbox(label="Example Input", visible=False, interactive=False)
         example_prediction = gr.Label(label="Model Sample Prediction", visible=False)
 
     with gr.Row():
@@ -148,6 +149,10 @@ def get_demo():
         fn=get_related_datasets_from_leaderboard,
         inputs=[model_id_input],
         outputs=[dataset_id_input],
+    ).then(
+        fn=check_dataset,
+        inputs=[dataset_id_input],
+        outputs=[dataset_config_input, dataset_split_input, loading_dataset_info],
     )
     
     gr.on(
@@ -202,6 +207,7 @@ def get_demo():
             example_btn, 
             first_line_ds,
             validation_result,
+            example_input,
             example_prediction,
             column_mapping_accordion,],
     )
@@ -222,6 +228,7 @@ def get_demo():
         ],
         outputs=[
             validation_result,
+            example_input,
             example_prediction,
             column_mapping_accordion,
             run_btn,

text_classification_ui_helpers.py

@@ -186,6 +186,7 @@ def precheck_model_ds_enable_example_btn(
             gr.update(visible=False),
             gr.update(visible=False),
             gr.update(visible=False),
+            gr.update(visible=False),
         )
 
     try:
@@ -198,7 +199,8 @@ def precheck_model_ds_enable_example_btn(
           return (
               gr.update(interactive=False), 
               gr.update(value=df, visible=True),
-              gr.update(visible=False), 
+              gr.update(visible=False),
+              gr.update(visible=False),
               gr.update(visible=False),
               gr.update(visible=False),
           )
@@ -211,6 +213,7 @@ def precheck_model_ds_enable_example_btn(
                 gr.update(visible=False),
                 gr.update(visible=False),
                 gr.update(visible=False),
+                gr.update(visible=False),
             )
 
         return (
@@ -219,12 +222,14 @@ def precheck_model_ds_enable_example_btn(
                 gr.update(visible=False),
                 gr.update(visible=False),
                 gr.update(visible=False),
+                gr.update(visible=False),
             )
     except Exception as e:
         # Config or split wrong
         logger.warn(f"Check your dataset {dataset_id} and config {dataset_config} on split {dataset_split}: {e}")
         return (
-            gr.update(interactive=False), 
+            gr.update(interactive=False),
+            gr.update(visible=False),
             gr.update(visible=False),
             gr.update(visible=False),
             gr.update(visible=False),
@@ -266,6 +271,7 @@ def align_columns_and_show_prediction(
 
     if prediction_input is None or prediction_response is None:
         return (
+            gr.update(visible=False),
             gr.update(visible=False),
             gr.update(visible=False),
             gr.update(visible=False, open=False),
@@ -276,6 +282,7 @@ def align_columns_and_show_prediction(
 
     if isinstance(prediction_response, HuggingFaceInferenceAPIResponse):
         return (
+            gr.update(visible=False),
             gr.update(visible=False),
             gr.update(visible=False),
             gr.update(visible=False, open=False),
@@ -293,6 +300,7 @@ def align_columns_and_show_prediction(
     if not isinstance(ds_labels, list) or not isinstance(ds_features, list):
         gr.Warning(CHECK_CONFIG_OR_SPLIT_RAW)
         return (
+            gr.update(visible=False),
             gr.update(visible=False),
             gr.update(visible=False),
             gr.update(visible=False, open=False),
@@ -305,6 +313,7 @@ def align_columns_and_show_prediction(
         return (
             gr.update(value=UNMATCHED_MODEL_DATASET_STYLED_ERROR, visible=True),
             gr.update(visible=False),
+            gr.update(visible=False),
             gr.update(visible=False, open=False),
             gr.update(interactive=False),
             "",
@@ -327,6 +336,7 @@ def align_columns_and_show_prediction(
         return (
             gr.update(value=MAPPING_STYLED_ERROR_WARNING, visible=True),
             gr.update(visible=False),
+            gr.update(visible=False),
             gr.update(visible=True, open=True),
             gr.update(interactive=(run_inference and inference_token != "")),
             "",
@@ -335,7 +345,8 @@ def align_columns_and_show_prediction(
 
     return (
         gr.update(value=VALIDATED_MODEL_DATASET_STYLED, visible=True),
-        gr.update(value=prediction_response, label=prediction_input, visible=True),
+        gr.update(value=prediction_input, lines=len(prediction_input)//225 + 1, visible=True),
+        gr.update(value=prediction_response, visible=True),
         gr.update(visible=True, open=False),
         gr.update(interactive=(run_inference and inference_token != "")),
         "",