SketchAgent with In-Context Learning

Forked from yael-vinker/sketch-agent

Setup

conda env create -f mac-environment.yml

Download dataset:

mkdir .data; cd .data; mkdir quickdraw; cd quickdraw; mkdir simplified; cd simplified;
gsutil -m cp -r gs://quickdraw_dataset/full/simplified/* .

Setup .env file:

Add your OpenAI API key to the .env file:

Usage

In Context Learning:

python icl_sketch.py --concept "airplane" --examples 3

Simple Sketch:

python simple_sketch.py --concept "airplane"

Read more about the DSL