Jingkang Yang commited on
Commit
500ca41
·
1 Parent(s): 9ab2d9a

update:requirments

Browse files
Files changed (2) hide show
  1. app.py +5 -1
  2. requirements.txt +0 -1
app.py CHANGED
@@ -1,6 +1,10 @@
1
  # Copyright (c) Facebook, Inc. and its affiliates.
2
  # Copyright (c) Meta Platforms, Inc. All Rights Reserved
3
-
 
 
 
 
4
  import argparse
5
  import glob
6
  import multiprocessing as mp
 
1
  # Copyright (c) Facebook, Inc. and its affiliates.
2
  # Copyright (c) Meta Platforms, Inc. All Rights Reserved
3
+ try:
4
+ import detectron2
5
+ except:
6
+ import os
7
+ os.system('pip install git+https://github.com/facebookresearch/detectron2.git')
8
  import argparse
9
  import glob
10
  import multiprocessing as mp
requirements.txt CHANGED
@@ -15,5 +15,4 @@ pytorch==1.10.1
15
  torchvision==0.11.2
16
  torchaudio==0.10.1
17
  cudatoolkit==11.3
18
- git+https://github.com/facebookresearch/detectron2.git
19
  git+https://github.com/openai/CLIP.git
 
15
  torchvision==0.11.2
16
  torchaudio==0.10.1
17
  cudatoolkit==11.3
 
18
  git+https://github.com/openai/CLIP.git