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()
|