Nolwenn
commited on
Commit
·
f5221b6
1
Parent(s):
d542bea
Update model caching
Browse files- crs_arena/arena.py +2 -3
crs_arena/arena.py
CHANGED
@@ -60,6 +60,8 @@ logger.setLevel(logging.INFO)
|
|
60 |
if not os.path.exists("data/models"):
|
61 |
logger.info("Downloading models...")
|
62 |
download_and_extract_models()
|
|
|
|
|
63 |
if not os.path.exists("data/embed_items"):
|
64 |
logger.info("Downloading item embeddings...")
|
65 |
download_and_extract_item_embeddings()
|
@@ -209,9 +211,6 @@ def feedback_dialog(row_id: int) -> None:
|
|
209 |
# Streamlit app
|
210 |
st.set_page_config(page_title="CRS Arena", layout="wide")
|
211 |
|
212 |
-
# Cache some CRS fighters at startup
|
213 |
-
cache_fighters()
|
214 |
-
|
215 |
# Battle setup
|
216 |
if "user_id" not in st.session_state:
|
217 |
st.session_state["user_id"] = get_unique_user_id()
|
|
|
60 |
if not os.path.exists("data/models"):
|
61 |
logger.info("Downloading models...")
|
62 |
download_and_extract_models()
|
63 |
+
# Cache some CRS fighters at startup
|
64 |
+
cache_fighters()
|
65 |
if not os.path.exists("data/embed_items"):
|
66 |
logger.info("Downloading item embeddings...")
|
67 |
download_and_extract_item_embeddings()
|
|
|
211 |
# Streamlit app
|
212 |
st.set_page_config(page_title="CRS Arena", layout="wide")
|
213 |
|
|
|
|
|
|
|
214 |
# Battle setup
|
215 |
if "user_id" not in st.session_state:
|
216 |
st.session_state["user_id"] = get_unique_user_id()
|