clean up and change run btn|

app_text_classification.py

@@ -11,7 +11,7 @@ from transformers.pipelines import TextClassificationPipeline
 
 from text_classification import get_labels_and_features_from_dataset, check_model, get_example_prediction, check_column_mapping_keys_validity, text_classification_fix_column_mapping
 from utils import read_scanners, write_scanners, read_inference_type, read_column_mapping, write_column_mapping, write_inference_type, convert_column_mapping_to_json
-from wordings import CONFIRM_MAPPING_DETAILS_MD, CONFIRM_MAPPING_DETAILS_FAIL_MD
+from wordings import CONFIRM_MAPPING_DETAILS_MD, CONFIRM_MAPPING_DETAILS_FAIL_MD, CONFIRM_MAPPING_DETAILS_FAIL_RAW
 
 HF_REPO_ID = 'HF_REPO_ID'
 HF_SPACE_ID = 'SPACE_ID'
@@ -28,15 +28,15 @@ def try_submit(m_id, d_id, config, split, local):
     all_mappings = read_column_mapping(CONFIG_PATH)
 
     if "labels" not in all_mappings.keys():
-        return gr.update(interactive=False)
+        gr.Warning(CONFIRM_MAPPING_DETAILS_FAIL_RAW)
+        return gr.update(interactive=True)
     label_mapping = all_mappings["labels"]
     
     if "features" not in all_mappings.keys():
-        return gr.update(interactive=False)
+        gr.Warning(CONFIRM_MAPPING_DETAILS_FAIL_RAW)
+        return gr.update(interactive=True)
     feature_mapping = all_mappings["features"]
 
-    print(label_mapping, feature_mapping)
-
     # TODO: Set column mapping for some dataset such as `amazon_polarity`
     if local:
         command = [
@@ -225,18 +225,8 @@ def get_demo():
     
     gr.on(
         triggers=[
-            model_id_input.change, 
-            dataset_id_input.change, 
-            dataset_config_input.change, 
-            dataset_split_input.change,
             run_btn.click,
             ],
         fn=try_submit,
         inputs=[model_id_input, dataset_id_input, dataset_config_input, dataset_split_input, run_local],
-        outputs=[run_btn])
-    
-    # gr.on(
-    #     triggers=[label.change for label in column_mappings],
-    #     fn=try_submit,
-    #     inputs=[model_id_input, dataset_id_input, dataset_config_input, dataset_split_input, run_local],
-    #     outputs=[run_btn])
+        outputs=[run_btn])

wordings.py

@@ -10,4 +10,8 @@ CONFIRM_MAPPING_DETAILS_FAIL_MD = '''
                             Confirm Pre-processing Details
                             </h1>
                             Sorry, we cannot align the input/output of your dataset with the model. <b>Pleaser double check your model and dataset.</b>
+                            '''
+
+CONFIRM_MAPPING_DETAILS_FAIL_RAW= '''
+                            Sorry, we cannot align the input/output of your dataset with the model. Pleaser double check your model and dataset.
                             '''