[Bugfix] .

fl0wxr · fl0wxr · commit a24ab6a0ac3b · 2024-04-16T12:46:07.000+03:00
diff --git a/python/main.py b/python/main.py
@@ -330,7 +330,7 @@ def load_model() -> MedsamLite:
         mask_decoder=medsam_lite_mask_decoder,
         prompt_encoder=medsam_lite_prompt_encoder,
     )
-    medsam_dir = Path(__file__).parent / "templates" / "lite_medsam.pth"
+    medsam_dir = Path(__file__).parent / "prm" / "lite_medsam.pth"
     medsam_lite_checkpoint = torch.load(medsam_dir, map_location="cpu")
     medsam_model.load_state_dict(medsam_lite_checkpoint)
     medsam_model.to("cpu")

Original file line number	Diff line number	Diff line change
`@@ -330,7 +330,7 @@ def load_model() -> MedsamLite:`
`330`	`330`	`mask_decoder=medsam_lite_mask_decoder,`
`331`	`331`	`prompt_encoder=medsam_lite_prompt_encoder,`
`332`	`332`	`)`
`333`		`- medsam_dir = Path(__file__).parent / "templates" / "lite_medsam.pth"`
	`333`	`+ medsam_dir = Path(__file__).parent / "prm" / "lite_medsam.pth"`
`334`	`334`	`medsam_lite_checkpoint = torch.load(medsam_dir, map_location="cpu")`
`335`	`335`	`medsam_model.load_state_dict(medsam_lite_checkpoint)`
`336`	`336`	`medsam_model.to("cpu")`