Spaces:

aadarsh99
/

ConverSeg

Running on Zero

aadarsh99 commited on 21 days ago

Commit

37ff04d

1 Parent(s): b485573

update model path

Files changed (1) hide show

app.py CHANGED Viewed

@@ -26,8 +26,8 @@ logging.basicConfig(level=logging.INFO)
 REPO_ID = "aadarsh99/ConvSeg-Stage2"
 SAM2_CONFIG = "sam2_hiera_l.yaml"
 BASE_CKPT_NAME = "sam2_hiera_large.pt"
-FINE_TUNED_SAM = "fine_tuned_sam2_batched_18000.torch"
-FINE_TUNED_PLM = "fine_tuned_sam2_batched_plm_18000.torch"
 SQUARE_DIM = 1024

 REPO_ID = "aadarsh99/ConvSeg-Stage2"
 SAM2_CONFIG = "sam2_hiera_l.yaml"
 BASE_CKPT_NAME = "sam2_hiera_large.pt"
+FINE_TUNED_SAM = "fine_tuned_sam2_batched_90000.torch"
+FINE_TUNED_PLM = "fine_tuned_sam2_batched_plm_90000.torch"
 SQUARE_DIM = 1024