Improve wordings for non-existing IDs

text_classification_ui_helpers.py

@@ -28,6 +28,7 @@ from wordings import (
     CHECK_CONFIG_OR_SPLIT_RAW,
     CONFIRM_MAPPING_DETAILS_FAIL_RAW,
     MAPPING_STYLED_ERROR_WARNING,
+    NOT_FOUND_DATASET_RAW,
     NOT_FOUND_MODEL_RAW,
     NOT_TEXT_CLASSIFICATION_MODEL_RAW,
     UNMATCHED_MODEL_DATASET_STYLED_ERROR,
@@ -91,10 +92,15 @@ def check_dataset(dataset_id):
         )
     except Exception as e:
         logger.warning(f"Check your dataset {dataset_id}: {e}")
-        if "doesn't exist" in str(e):
+        if "doesn't exist on the Hub or cannot be accessed" in str(e):
+            gr.Warning(NOT_FOUND_DATASET_RAW)
+        elif "forbidden" in str(e).lower():
+            # GSK-2770: illegal name
             gr.Warning(get_dataset_fetch_error_raw(e))
-        if "forbidden" in str(e).lower():  # GSK-2770
+        else:
+            # Unknown error
             gr.Warning(get_dataset_fetch_error_raw(e))
+
         return (gr.update(visible=False), gr.update(visible=False), "")
 
 

wordings.py

@@ -73,7 +73,7 @@ UNMATCHED_MODEL_DATASET_STYLED_ERROR = """
                         """
 
 NOT_FOUND_MODEL_RAW = """
-    We cannot find your model on Hugging Face. Please ensure that the model is accessible.
+    We cannot find your model on Hugging Face. Please check that the model id is correct and publicly accessible.
 """
 
 NOT_TEXT_CLASSIFICATION_MODEL_RAW = """
@@ -111,5 +111,9 @@ VALIDATED_MODEL_DATASET_STYLED = """
             </h3>"""
 
 
+NOT_FOUND_DATASET_RAW = """
+    We cannot find your dataset on Hugging Face. Please check that the dataset id is correct and publicly accessible.
+"""
+
 def get_dataset_fetch_error_raw(error):
-    return f"""Sorry you cannot use this dataset because {error}. Contact HF team to support this dataset."""
+    return f"""Sorry you cannot use this dataset because of an unknwon error: "{error}"."""