- groundingdino_swint_ogc.pth +3 -0
- handler.py +2 -0
groundingdino_swint_ogc.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3b3ca2563c77c69f651d7bd133e97139c186df06231157a64c507099c52bc799
|
| 3 |
+
size 693997677
|
handler.py
CHANGED
|
@@ -27,6 +27,8 @@ class EndpointHandler():
|
|
| 27 |
|
| 28 |
# Install Grounding DINO
|
| 29 |
subprocess.run(["python", "-m", "pip", "install", "-e", "GroundingDINO"])
|
|
|
|
|
|
|
| 30 |
|
| 31 |
# Install diffusers
|
| 32 |
subprocess.run(["pip", "install", "--upgrade", "diffusers[torch]"])
|
|
|
|
| 27 |
|
| 28 |
# Install Grounding DINO
|
| 29 |
subprocess.run(["python", "-m", "pip", "install", "-e", "GroundingDINO"])
|
| 30 |
+
|
| 31 |
+
subprocess.run("wget https://huggingface.co/Uminosachi/sam-hq/resolve/main/sam_hq_vit_h.pth -O ./sam_hq_vit_h.pth", shell=True)
|
| 32 |
|
| 33 |
# Install diffusers
|
| 34 |
subprocess.run(["pip", "install", "--upgrade", "diffusers[torch]"])
|