Create app.py

app.py

@@ -0,0 +1,63 @@
+import os
+import requests
+from tqdm import tqdm
+import logging
+
+def download_sam_model(url="https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth", 
+                      save_dir="."):
+    """
+    Download the SAM model weights to the root directory with progress tracking.
+    
+    Args:
+        url (str): URL of the model weights
+        save_dir (str): Directory to save the downloaded file (defaults to current directory)
+    """
+    try:
+        # Extract filename from URL
+        filename = url.split('/')[-1]
+        save_path = os.path.join(save_dir, filename)
+        
+        # Setup logging
+        logging.basicConfig(level=logging.INFO)
+        logger = logging.getLogger(__name__)
+        
+        # Check if file already exists
+        if os.path.exists(save_path):
+            logger.info(f"File {filename} already exists in {save_dir}")
+            return save_path
+            
+        # Send a HEAD request to get the file size
+        response = requests.head(url)
+        file_size = int(response.headers.get('content-length', 0))
+        
+        # Download the file with progress bar
+        logger.info(f"Downloading {filename} to root directory")
+        response = requests.get(url, stream=True)
+        progress = tqdm(total=file_size, unit='iB', unit_scale=True)
+        
+        with open(save_path, 'wb') as file:
+            for data in response.iter_content(chunk_size=1024):
+                progress.update(len(data))
+                file.write(data)
+        
+        progress.close()
+        
+        if file_size != 0 and progress.n != file_size:
+            logger.error("Error during download - incomplete file")
+            raise Exception("Downloaded file size does not match expected size")
+            
+        logger.info(f"Successfully downloaded {filename}")
+        return save_path
+        
+    except Exception as e:
+        logger.error(f"Error downloading file: {str(e)}")
+        raise
+        
+if __name__ == "__main__":
+    MODEL_URL = "https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth"
+    
+    try:
+        downloaded_path = download_sam_model(MODEL_URL)
+        print(f"Model downloaded successfully to: {downloaded_path}")
+    except Exception as e:
+        print(f"Failed to download model: {str(e)}")