Spaces:
Sleeping
Sleeping
efengx
commited on
Commit
·
ae06165
1
Parent(s):
8129782
fix: update
Browse files- fengxai/inference.py +3 -0
- requirements.txt +0 -1
fengxai/inference.py
CHANGED
@@ -4,6 +4,9 @@ from fengxai.SelfSupervised import selfSupervised
|
|
4 |
def imagePrediction(imageFile, funcCallback):
|
5 |
if not os.path.exists("weights/sam_vit_h_4b8939.pth") \
|
6 |
and not os.path.exists("weights/groundingdino_swint_ogc.pth"):
|
|
|
|
|
|
|
7 |
cmd = """
|
8 |
mkdir weights \
|
9 |
&& cd weights \
|
|
|
4 |
def imagePrediction(imageFile, funcCallback):
|
5 |
if not os.path.exists("weights/sam_vit_h_4b8939.pth") \
|
6 |
and not os.path.exists("weights/groundingdino_swint_ogc.pth"):
|
7 |
+
cmd = "pip install git+https://github.com/IDEA-Research/GroundingDINO.git"
|
8 |
+
os.system(cmd)
|
9 |
+
|
10 |
cmd = """
|
11 |
mkdir weights \
|
12 |
&& cd weights \
|
requirements.txt
CHANGED
@@ -2,7 +2,6 @@ streamlit
|
|
2 |
Pillow
|
3 |
segment_anything
|
4 |
diffusers
|
5 |
-
git+https://github.com/IDEA-Research/GroundingDINO.git
|
6 |
scipy
|
7 |
torch
|
8 |
torchvision
|
|
|
2 |
Pillow
|
3 |
segment_anything
|
4 |
diffusers
|
|
|
5 |
scipy
|
6 |
torch
|
7 |
torchvision
|