launch app in the docker

Dockerfile

@@ -46,4 +46,6 @@ ARG IABN_FORCE_CUDA="1"
 ARG FORCE_CUDA="1"
 RUN pip3 install --no-cache-dir \
     inplace_abn \
-    git+https://github.com/mit-han-lab/[email protected]
+    git+https://github.com/mit-han-lab/[email protected]
+
+CMD ["python3", "app.py"]

app.py

@@ -11,16 +11,6 @@ sys.path.append(code_dir)
 elev_est_dir = os.path.abspath(os.path.join(code_dir, "elevation_estimate"))
 sys.path.append(elev_est_dir)
 
-# if not is_local_run:
-    # export TORCH_CUDA_ARCH_LIST="7.0;7.2;8.0;8.6"
-    # export IABN_FORCE_CUDA=1
-    # os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6"
-    # os.environ["IABN_FORCE_CUDA"] = "1"
-    # os.environ["FORCE_CUDA"] = "1"
-    # subprocess.run(['pip', 'install', 'inplace_abn'])
-    # FORCE_CUDA=1 pip install --no-cache-dir git+https://github.com/mit-han-lab/[email protected]
-    # subprocess.run(['pip', 'install', '--no-cache-dir', 'git+https://github.com/mit-han-lab/[email protected]'])
-
 import shutil
 import torch
 import fire