Spaces:

indigoman
/

text_image_editing

Runtime error

artyaltanzaya commited on Jul 31, 2023

Commit

7a2e256

1 Parent(s): 3f76825

app

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,11 +18,11 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 print("path", os.getcwd())
 model_type = "vit_h"
-predictor = SamPredictor(sam_model_registry[model_type](checkpoint="./weights/sam_vit_h_4b8939.pth").to(device))
 # ------Stable Diffusion
 pipe = StableDiffusionInpaintPipeline.from_pretrained("stabilityai/stable-diffusion-2-inpainting", torch_dtype=torch.float16,).to(device)
 # ----Grounding DINO
-groundingdino_model = load_model("./groundingdino/config/GroundingDINO_SwinT_OGC.py", "./weights/groundingdino_swint_ogc.pth")
 BOX_TRESHOLD = 0.3
 TEXT_TRESHOLD = 0.25

 print("path", os.getcwd())
 model_type = "vit_h"
+predictor = SamPredictor(sam_model_registry[model_type](checkpoint="./GroundingDINO/weights/sam_vit_h_4b8939.pth").to(device))
 # ------Stable Diffusion
 pipe = StableDiffusionInpaintPipeline.from_pretrained("stabilityai/stable-diffusion-2-inpainting", torch_dtype=torch.float16,).to(device)
 # ----Grounding DINO
+groundingdino_model = load_model("./GroundingDINO/groundingdino/config/GroundingDINO_SwinT_OGC.py", "./GroundingDINO/weights/groundingdino_swint_ogc.pth")
 BOX_TRESHOLD = 0.3
 TEXT_TRESHOLD = 0.25