import os
import requests

# The URL of the file you want to download
url = 'https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth'

# The path where you want to save the downloaded file
save_path = './models/sam_vit_h_4b8939.pth'

# Check if the file already exists
if not os.path.exists(save_path):
    # Send a GET request to the URL
    response = requests.get(url, stream=True)

    # Check if the request was successful
    if response.status_code == 200:
        # Open the file in write mode
        with open(save_path, 'wb') as file:
            # Write the contents of the response to the file
            for chunk in response.iter_content(chunk_size=1024):
                if chunk:
                    file.write(chunk)
else:
    print("The file already exists.")