fix markdown load html freeze bug

app_text_classification.py

@@ -42,7 +42,7 @@ def get_demo(demo):
         dataset_split_input = gr.Dropdown(label="Dataset Split", visible=False)
 
     with gr.Row():
-        example_input = gr.Markdown("Example Input", visible=False)
+        example_input = gr.HTML(visible=False)
     with gr.Row():
         example_prediction = gr.Label(label="Model Prediction Sample", visible=False)
 

text_classification_ui_helpers.py

@@ -7,6 +7,7 @@ import threading
 import datasets
 import gradio as gr
 from transformers.pipelines import TextClassificationPipeline
+from wordings import get_styled_input
 
 from io_utils import (get_yaml_path, read_column_mapping, save_job_to_pipe,
                       write_column_mapping, write_inference_type,
@@ -181,7 +182,7 @@ def check_model_and_show_prediction(
         ppl, dataset_id, dataset_config, dataset_split
     )
     return (
-        gr.update(value=prediction_input, visible=True),
+        gr.update(value=get_styled_input(prediction_input), visible=True),
         gr.update(value=prediction_output, visible=True),
         gr.update(visible=True, open=False),
         *column_mappings,

wordings.py

@@ -37,3 +37,8 @@ MAPPING_STYLED_ERROR_WARNING = """
                         Sorry, we cannot auto-align the labels/features of your dataset and model. Please double check.
                         </h3>
                         """
+
+def get_styled_input(input):
+    return f"""<h3 style="text-align: center;color: #5ec26a; background-color: #e2fbe8; border-radius: 8px; padding: 10px; ">
+            Sample input: {input}
+            </h3>"""