laizeqiang commited on
Commit
b49eb8e
1 Parent(s): f2e1adc
Files changed (2) hide show
  1. iChatApp.py +5 -0
  2. requirements.txt +1 -2
iChatApp.py CHANGED
@@ -6,6 +6,11 @@ from pathlib import Path
6
  import sys
7
  sys.path.insert(0, str(Path(__file__).resolve().parent / "third-party" / "lama"))
8
 
 
 
 
 
 
9
  import random
10
  import torch
11
  import cv2
 
6
  import sys
7
  sys.path.insert(0, str(Path(__file__).resolve().parent / "third-party" / "lama"))
8
 
9
+ try:
10
+ import detectron
11
+ except:
12
+ os.system('pip install git+https://github.com/facebookresearch/detectron2.git')
13
+
14
  import random
15
  import torch
16
  import cv2
requirements.txt CHANGED
@@ -22,5 +22,4 @@ webdataset==0.2.48
22
  kornia==0.5.0
23
  sentencepiece==0.1.98
24
  accelerate==0.18.0
25
- git+https://github.com/facebookresearch/segment-anything.git
26
- git+https://github.com/facebookresearch/detectron2.git
 
22
  kornia==0.5.0
23
  sentencepiece==0.1.98
24
  accelerate==0.18.0
25
+ git+https://github.com/facebookresearch/segment-anything.git