wrap hf dataset error

text_classification_ui_helpers.py

@@ -26,6 +26,7 @@ from wordings import (
     UNMATCHED_MODEL_DATASET_STYLED_ERROR,
     CHECK_LOG_SECTION_RAW,
     get_styled_input,
+    get_dataset_fetch_error_raw,
 )
 import os
 
@@ -70,6 +71,8 @@ def check_dataset(dataset_id):
         )
     except Exception as e:
         logger.warn(f"Check your dataset {dataset_id}: {e}")
+        if "forbidden" in str(e).lower(): # GSK-2770
+            gr.warning(get_dataset_fetch_error_raw(e))
         return (
             gr.update(),
             gr.update(),

wordings.py

@@ -77,6 +77,9 @@ HF_TOKEN_INVALID_STYLED= """
                 </p>
                 """
 
+def get_dataset_fetch_error_raw(error):
+    return f"""Sorry you cannot use this dataset because {error} Contact HF team to support this dataset."""
+
 def get_styled_input(input):
     return f"""<h3 style="text-align: center;color: #4ca154; background-color: #e2fbe8; border-radius: 8px; padding: 10px; ">
             Your model and dataset have been validated! <br /> Sample input: {input}