laizeqiang
commited on
Commit
•
b49eb8e
1
Parent(s):
f2e1adc
update
Browse files- iChatApp.py +5 -0
- 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
|
|