update: support scannet

app.py

@@ -1,18 +1,18 @@
 # Copyright (c) Facebook, Inc. and its affiliates.
 # Copyright (c) Meta Platforms, Inc. All Rights Reserved
 
-# import os
-# os.system('pip install torch==1.10.1+cu113 torchvision==0.11.2+cu113 torchaudio==0.10.1+cu113 -f https://download.pytorch.org/whl/torch_stable.html')
+import os
+os.system('pip install torch==1.10.1+cu113 torchvision==0.11.2+cu113 torchaudio==0.10.1+cu113 -f https://download.pytorch.org/whl/torch_stable.html')
 
-# try:
-#     import detectron2
-# except:
-#     import os 
-#     # os.system('cd /home/user/app/third_party/CLIP && pip install -Ue .')
-#     os.system('pip install git+https://github.com/Jun-CEN/CLIP.git')
-#     os.system('pip install git+https://github.com/facebookresearch/detectron2.git')
-#     os.system('pip install git+https://github.com/facebookresearch/pytorch3d.git')
-#     os.system('pip install git+https://github.com/facebookresearch/segment-anything.git')
+try:
+    import detectron2
+except:
+    import os 
+    # os.system('cd /home/user/app/third_party/CLIP && pip install -Ue .')
+    os.system('pip install git+https://github.com/Jun-CEN/CLIP.git')
+    os.system('pip install git+https://github.com/facebookresearch/detectron2.git')
+    os.system('pip install git+https://github.com/facebookresearch/pytorch3d.git')
+    os.system('pip install git+https://github.com/facebookresearch/segment-anything.git')
     
 import argparse
 import glob