sketchkit.sketch2image.nano_banana2 package

Module contents

Nano Banana 2 - Google Gemini API for sketch-to-image conversion.

class sketchkit.sketch2image.nano_banana2.NanoBanana2Model(device: str = 'cuda', model_name: str = 'gemini-3.1-flash-image-preview', **kwargs)[source]

Bases: object

Nano Banana 2 model using Google Gemini API.

Note

API key must be set in ~/.sketchkit/config as GOOGLE_API_KEY=”your_api_key”

generate(control_image: Image, prompt: str, size: int | Tuple[int, int] | None = None, negative_prompt: str | None = None, **kwargs) Image[source]

Generate image from sketch using Google Gemini API.