test_endpoint.py

from handler import EndpointHandler
import os

# Create results directory
os.makedirs("./assets", exist_ok=True)

# init handler
my_handler = EndpointHandler(path=".")

# prepare sample prompts
prompts = [
    "A dynamic city skyline, stylized and white-hued, stretches across the midsection, featuring abstract buildings and architectural shapes against a white background. The top section reveals a person composing a photograph on a bridge, providing a sense of scale and adventure. The image carries black and white elements, with the city block emerging prominently from the vivid environment, blending creativity and urban aesthetics.",
    "A digital artwork showcases a futuristic cityscape at dusk, viewed through a wide-angle lens. The buildings are sleek and towering, reflecting a mix of modern and cyberpunk architecture, set against a vivid sunset. Swirls of cloud and city lights create a dynamic, lively atmosphere. The image is bordered with intricate orange and black geometric patterns, enhancing contemporary aesthetic against a plain white background.",
    "[women tshirts][back] The t-shirt features a soft, pastel pink background texture, providing a calm and dreamy aesthetic. The graphic design on the shirt includes a light blue suitcase with whimsical details such as a rainbow, stars, and a drifting cloud, adding a fantasy element. This is further accented by pastel-colored items like a star and a small bag, enhancing the dreamlike theme."
]

# Generate and save images
for i, prompt in enumerate(prompts, 1):
    image = my_handler({"inputs": prompt})
    image.save(f"./results/cityscape_{i}.png")
    print(image)
    #print(f"Saved image {i} to ./results/cityscape_{i}.png")

# non_holiday_pred [{'label': 'joy', 'score': 0.9985942244529724}]
# holiday_payload [{'label': 'happy', 'score': 1}]