from PIL import Image, ImageDraw
from sentence_transformers import SentenceTransformer


MODEL_ID = "sentence-transformers/clip-ViT-B-32"
TEXT_LABELS = ["a red square", "a blue circle", "a green triangle"]

image = Image.new("RGB", (224, 224), "white")
draw = ImageDraw.Draw(image)
draw.rectangle((40, 40, 184, 184), fill=(220, 30, 30))