Nolwenn
commited on
Commit
·
95b7f82
1
Parent(s):
5caf0ed
Update Dockerfile
Browse files- Dockerfile +1 -1
- download_external_data.py +5 -0
Dockerfile
CHANGED
@@ -16,4 +16,4 @@ RUN python download_external_data.py
|
|
16 |
EXPOSE 7860
|
17 |
|
18 |
# Run Streamlit app
|
19 |
-
CMD ["python", "-m", "streamlit", "run", "crs_arena.
|
|
|
16 |
EXPOSE 7860
|
17 |
|
18 |
# Run Streamlit app
|
19 |
+
CMD ["python", "-m", "streamlit", "run", "crs_arena/arena.py", "--server.port", "7860"]
|
download_external_data.py
CHANGED
@@ -14,6 +14,7 @@ def download_and_extract_models() -> None:
|
|
14 |
models_targz = "models.tar.gz"
|
15 |
models_folder = "data/models/"
|
16 |
try:
|
|
|
17 |
wget.download(models_url, models_targz)
|
18 |
|
19 |
logging.debug("Extracting models folder.")
|
@@ -34,6 +35,9 @@ def download_and_extract_item_embeddings() -> None:
|
|
34 |
item_embeddings_folder = "data/"
|
35 |
|
36 |
try:
|
|
|
|
|
|
|
37 |
wget.download(item_embeddings_url, item_embeddings_tarbz)
|
38 |
|
39 |
logging.debug("Extracting item embeddings folder.")
|
@@ -47,6 +51,7 @@ def download_and_extract_item_embeddings() -> None:
|
|
47 |
|
48 |
|
49 |
if __name__ == "__main__":
|
|
|
50 |
if not os.path.exists("data/models"):
|
51 |
logging.info("Downloading models...")
|
52 |
download_and_extract_models()
|
|
|
14 |
models_targz = "models.tar.gz"
|
15 |
models_folder = "data/models/"
|
16 |
try:
|
17 |
+
logging.debug(f"Downloading models from {models_url}.")
|
18 |
wget.download(models_url, models_targz)
|
19 |
|
20 |
logging.debug("Extracting models folder.")
|
|
|
35 |
item_embeddings_folder = "data/"
|
36 |
|
37 |
try:
|
38 |
+
logging.debug(
|
39 |
+
f"Downloading item embeddings from {item_embeddings_url}."
|
40 |
+
)
|
41 |
wget.download(item_embeddings_url, item_embeddings_tarbz)
|
42 |
|
43 |
logging.debug("Extracting item embeddings folder.")
|
|
|
51 |
|
52 |
|
53 |
if __name__ == "__main__":
|
54 |
+
logging.basicConfig(level=logging.DEBUG)
|
55 |
if not os.path.exists("data/models"):
|
56 |
logging.info("Downloading models...")
|
57 |
download_and_extract_models()
|