Spaces:

NeoLee
/

DINO

Sleeping

NeoLee commited on Aug 6, 2024

Commit

5a3fb4e

verified ·

1 Parent(s): 71c5e40

update app

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ def dino_inference(image, text):
     model = load_model("groundingdino/config/GroundingDINO_SwinT_OGC.py",
     "weights/groundingdino_swint_ogc.pth")
-    IMAGE_PATH = "/teamspace/studios/this_studio/" + new_path
     TEXT_PROMPT = text # "apple . fruit basket . table ."
     BOX_TRESHOLD = 0.35
     TEXT_TRESHOLD = 0.25

     model = load_model("groundingdino/config/GroundingDINO_SwinT_OGC.py",
     "weights/groundingdino_swint_ogc.pth")
+    IMAGE_PATH = new_path
     TEXT_PROMPT = text # "apple . fruit basket . table ."
     BOX_TRESHOLD = 0.35
     TEXT_TRESHOLD = 0.25