clip_interrogator_image_to_text_prompt_generation.py

1from PIL import Image
2from clip_interrogator import Config, Interrogator
3
4# Load the image you want to interrogate
5image_path = "image.jpg"
6image = Image.open(image_path).convert('RGB')
7
8# Initialize the interrogator with default configurations
9# This will download the necessary CLIP and BLIP models on first run
10ci = Interrogator(Config(clip_model_name="ViT-L-14/openai"))
11
12# Generate the prompt
13result = ci.interrogate(image)
14
15print(result)