Astria API CLI client and SDK

This is an simple python SDK and CLI for Astria.ai, the DreamBooth based image generation service.

CLI usage

When used as a command line tool, these are the arguments:

Suppored actions:

gen                 Generate images for prompts
tune                Create a new tune
list                List all generated images for a tune
listtune            List all tunes in the account
info                Get details of prompt
download            Download images of prompt
downloadall         Download recent images

Optional arguments:

• --key KEY Astria API token (if unspecified, token is taken from the enviroment variable ASTRIA_API_TOKEN)
• --outdir OUTDIR Output directory for images (default is current working directory)

Usage for gen action:

python astria.py gen [--seed SEED] [--steps STEPS] [--download] tuneid prompts [prompts ...]

• --seed SEED
• --steps STEPS
• --download Download results

Usage for tune action:

python astria.py tune [--test] title classname images [images ...]

title       Descriptive name for this tuning
classname   Dreambooth classname (style, person, man, woman etc)
images      At least 10 image filenames

Usage for list action:

python astria.py list tuneid

Usage for listtune action:

python astria.py listtune

Usage for info action:

python astria.py info tuneid promptid

• tuneid Astria Tune ID
• promptid Prompt ID

Usage for download action:

python astria.py download [--wait] tuneid promptid

• tuneid Astria Tune ID

• promptid Prompt ID

  optional arguments:

• --wait Wait for results to be ready

Usage for downloadall action:

python astria.py downloadall [--limit LIMIT] [--db DB] [tuneids ...]

• tuneids One or more Astria Tune IDs, or emtpy to download entire account

  optional arguments:

• --limit LIMIT Maximum number of prompts to download from each fine tune

• --db DB Keep list of previously downloaded images in a DB file