text-to-image / utils /image_utils.py
MA29's picture
first commit
f40a9c3
Raw History Blame Contribute Delete
315 Bytes
import os
from datetime import datetime
OUTPUT_DIR = "outputs/generated_images"
os.makedirs(OUTPUT_DIR, exist_ok=True)
def save_image(image):
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
path = os.path.join(OUTPUT_DIR, f"generated_{timestamp}.png")
image.save(path)
return path