Nolwenn
commited on
Commit
·
4f5b924
1
Parent(s):
37e134d
Update loading message
Browse files- crs_arena/utils.py +5 -4
crs_arena/utils.py
CHANGED
@@ -21,12 +21,13 @@ from src.model.crs_model import CRSModel
|
|
21 |
|
22 |
# Initialize Hugging Face API
|
23 |
HF_API = HfApi(token=os.environ.get("hf_token"))
|
|
|
|
|
|
|
|
|
24 |
|
25 |
|
26 |
-
@st.cache_resource(
|
27 |
-
show_spinner="The fighters are warming up... :robot: :punch: :gun: :boom:",
|
28 |
-
ttl=timedelta(days=3),
|
29 |
-
)
|
30 |
def get_crs_model(model_name: str, model_config_file: str) -> CRSModel:
|
31 |
"""Returns a CRS model.
|
32 |
|
|
|
21 |
|
22 |
# Initialize Hugging Face API
|
23 |
HF_API = HfApi(token=os.environ.get("hf_token"))
|
24 |
+
LOADING_MESSAGE = (
|
25 |
+
"The fighters are warming up, please be patient it may take a moment... "
|
26 |
+
":robot_face: :punch: :gun: :boom:"
|
27 |
+
)
|
28 |
|
29 |
|
30 |
+
@st.cache_resource(show_spinner=LOADING_MESSAGE, ttl=timedelta(days=3))
|
|
|
|
|
|
|
31 |
def get_crs_model(model_name: str, model_config_file: str) -> CRSModel:
|
32 |
"""Returns a CRS model.
|
33 |
|