Nolwenn commited on
Commit
95b7f82
·
1 Parent(s): 5caf0ed

Update Dockerfile

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. 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.arena", "--server.port", "7860"]
 
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()