Text-to-Image using Replicate

Install Dependencies

!uv pip install replicate -q

Set Replicate API Token

import sys
import os
from dotenv import load_dotenv

if 'google.colab' in sys.modules:
  from google.colab import userdata # type:ignore
  os.environ['REPLICATE_API_TOKEN'] = userdata.get('REPLICATE_API_TOKEN')
  print("Replicate API Token set for Colab")
else:
  load_dotenv()
  print("Loaded env vars from .env")

Text-to-Image

import replicate
output = replicate.run(
  "black-forest-labs/flux-pro",
  input={
      "steps": 28,
      "prompt": "lemon cupcake spelling out the words 'DigiPen' with sparklers, tasty, food photography, dynamic shot",
      "seed": 1564435,
      "output_format": "png",
      "safety_tolerance": 2,
      "prompt_upsampling": False
  },
)

import io
from PIL import Image

output_bytes = io.BytesIO(output.read())
output_image = Image.open(output_bytes)
output_image