Commit
•
873b855
1
Parent(s):
22bf258
Fix
Browse files- handler.py +5 -1
handler.py
CHANGED
|
@@ -4,8 +4,12 @@ from typing import Dict, List, Any
|
|
| 4 |
import groundingdino
|
| 5 |
from groundingdino.util.inference import load_model, load_image, predict, annotate
|
| 6 |
|
|
|
|
| 7 |
HOME = os.getcwd()
|
| 8 |
-
|
|
|
|
|
|
|
|
|
|
| 9 |
CONFIG_PATH = os.path.join(PACKAGE_HOME, "config", "GroundingDINO_SwinT_OGC.py")
|
| 10 |
WEIGHTS_PATH = os.path.join(HOME, "weights", "groundingdino_swint_ogc.pth")
|
| 11 |
|
|
|
|
| 4 |
import groundingdino
|
| 5 |
from groundingdino.util.inference import load_model, load_image, predict, annotate
|
| 6 |
|
| 7 |
+
# /app
|
| 8 |
HOME = os.getcwd()
|
| 9 |
+
|
| 10 |
+
# /opt/conda/lib/python3.9/site-packages/groundingdino
|
| 11 |
+
PACKAGE_HOME = os.path.dirname(groundingdino.__file__)
|
| 12 |
+
|
| 13 |
CONFIG_PATH = os.path.join(PACKAGE_HOME, "config", "GroundingDINO_SwinT_OGC.py")
|
| 14 |
WEIGHTS_PATH = os.path.join(HOME, "weights", "groundingdino_swint_ogc.pth")
|
| 15 |
|