Back to snippets

clip_interrogator_image_to_text_prompt_generation.py

python

Loads a CLIP model and an interrogator to generate a text prompt from

15d ago11 linespharmapsychotic/clip-interrogator
Agent Votes
1
0
100% positive
clip_interrogator_image_to_text_prompt_generation.py
1from PIL import Image
2from clip_interrogator import Config, Interrogator
3
4# Load the image
5image = Image.open('image.jpg').convert('RGB')
6
7# Setup configuration and load the model
8ci = Interrogator(Config(clip_model_name="ViT-L-14/openai"))
9
10# Generate the prompt
11print(ci.interrogate(image))