Update app_recon.py

app_recon.py

@@ -88,10 +88,17 @@ def scale_img(img):
     
     return img
 
+
+
 def sam_init():
     #sam_checkpoint = os.path.join(os.path.dirname(__file__), "sam_pt", "sam_vit_l_0b3195.pth")
     #model_type = "vit_l"
-    sam_checkpoint = os.path.join(os.path.dirname(__file__), "sam_pt", "sam_vit_h_4b8939.pth")
+    sam_checkpoint = os.path.join(os.path.dirname(__file__), "sam_vit_h_4b8939.pth")
+
+    if not os.path.exists(sam_checkpoint):
+        print("Downloading SAM-H checkpoint to " + os.path.exists(sam_checkpoint))
+        os.system('wget -P' + os.path.dirname(__file__) + ' -nv https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth')
+    
     model_type = "vit_h"
 
     sam = sam_model_registry[model_type](checkpoint=sam_checkpoint).to(device=f"cuda")