mirror of
https://github.com/brycedrennan/imaginAIry
synced 2024-10-31 03:20:40 +00:00
27 lines
718 B
Python
27 lines
718 B
Python
"""Command for generating image descriptions"""
|
|
|
|
import click
|
|
|
|
|
|
@click.argument("image_filepaths", nargs=-1)
|
|
@click.command()
|
|
def describe_cmd(image_filepaths):
|
|
"""Generate text descriptions of images."""
|
|
import os
|
|
|
|
from imaginairy.enhancers.describe_image_blip import generate_caption
|
|
from imaginairy.schema import LazyLoadingImage
|
|
|
|
imgs = []
|
|
for p in image_filepaths:
|
|
if p.startswith("http"):
|
|
img = LazyLoadingImage(url=p)
|
|
elif os.path.isdir(p):
|
|
print(f"Skipping directory: {p}")
|
|
continue
|
|
else:
|
|
img = LazyLoadingImage(filepath=p)
|
|
imgs.append(img)
|
|
for img in imgs:
|
|
print(generate_caption(img.copy()))
|