Error handling for incorrect repo id

start.py

@@ -22,6 +22,7 @@ import pandas as pd
 import sklearn
 import streamlit as st
 from huggingface_hub import hf_hub_download
+from huggingface_hub.utils import HFValidationError, RepositoryNotFoundError
 from sklearn.base import BaseEstimator
 from sklearn.dummy import DummyClassifier
 
@@ -126,8 +127,15 @@ def create_hf_model_card() -> None:
     if not repo_id:
         return
 
-    print("downloading model card")
-    path = hf_hub_download(repo_id, "README.md")
+    try:
+        path = hf_hub_download(repo_id, "README.md")
+    except (HFValidationError, RepositoryNotFoundError):
+        st.error(
+            f"Repository '{repo_id}' could not be found on HF Hub, "
+            "please check that the repo ID is correct."
+        )
+        return
+
     model_card = card.parse_modelcard(path)
     st.session_state.model_card = model_card
     st.session_state.model_card_type = "loaded"