Fix bug when loading model card with imgs

start.py

@@ -21,7 +21,7 @@ from tempfile import mkdtemp
 import pandas as pd
 import sklearn
 import streamlit as st
-from huggingface_hub import hf_hub_download
+from huggingface_hub import snapshot_download
 from huggingface_hub.utils import HFValidationError, RepositoryNotFoundError
 from sklearn.base import BaseEstimator
 from sklearn.dummy import DummyClassifier
@@ -98,7 +98,6 @@ def init_repo() -> None:
             data=data,
             requirements=requirements,
         )
-        1
     except Exception as exc:
         print("Uh oh, something went wrong when initializing the repo:", exc)
 
@@ -130,7 +129,7 @@ def create_hf_model_card() -> None:
         return
 
     try:
-        path = hf_hub_download(repo_id, "README.md")
+        path = snapshot_download(repo_id)
     except (HFValidationError, RepositoryNotFoundError):
         st.error(
             f"Repository '{repo_id}' could not be found on HF Hub, "
@@ -138,7 +137,12 @@ def create_hf_model_card() -> None:
         )
         return
 
-    model_card = card.parse_modelcard(path)
+    # move everything to the hf_path and working dir
+    hf_path = st.session_state.hf_path
+    shutil.copytree(path, hf_path, dirs_exist_ok=True)
+    shutil.copytree(path, ".", dirs_exist_ok=True)
+
+    model_card = card.parse_modelcard(hf_path / "README.md")
     st.session_state.model_card = model_card
     st.session_state.model_card_type = "loaded"
     st.session_state.screen.state = "edit"