efengx commited on
Commit
ae06165
·
1 Parent(s): 8129782

fix: update

Browse files
Files changed (2) hide show
  1. fengxai/inference.py +3 -0
  2. 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