memechain/tools/download_image.py

15 lines
443 B
Python

import os
import requests
def download_image(url):
response = requests.get(url)
if response.status_code == 200:
output_dir = 'output'
os.makedirs(output_dir, exist_ok=True)
image_path = os.path.join(output_dir, url.split("/")[-1])
with open(image_path, 'wb') as f:
f.write(response.content)
print(f"Image saved to {image_path}")
else:
print("Failed to download image.")