Jingkang Yang commited on
Commit
64aa48b
·
1 Parent(s): 4bcc656

update: clip

Browse files
Files changed (2) hide show
  1. app.py +6 -0
  2. requirements.txt +1 -4
app.py CHANGED
@@ -1,5 +1,11 @@
1
  # Copyright (c) Facebook, Inc. and its affiliates.
2
  # Copyright (c) Meta Platforms, Inc. All Rights Reserved
 
 
 
 
 
 
3
  try:
4
  import detectron2
5
  except:
 
1
  # Copyright (c) Facebook, Inc. and its affiliates.
2
  # Copyright (c) Meta Platforms, Inc. All Rights Reserved
3
+ try:
4
+ import torch
5
+ except:
6
+ import os
7
+ os.system('pip installtorch==1.10.1+cu113 torchvision==0.11.2+cu113 torchaudio==0.10.1+cu113 -f https://download.pytorch.org/whl/torch_stable.html')
8
+
9
  try:
10
  import detectron2
11
  except:
requirements.txt CHANGED
@@ -9,7 +9,4 @@ opencv-python
9
  pandas
10
  imageio
11
  fvcore
12
- iopath
13
- torch==1.10.1+cu113
14
- torchvision==0.11.2+cu113
15
- torchaudio==0.10.1+cu113
 
9
  pandas
10
  imageio
11
  fvcore
12
+ iopath