Replicate - Run AI with an API

# Run AI with an API Run and fine-tune open-source models. Deploy custom models at scale. All with one line of code. ## What You Can Do With Replicate - Generate images - Generate text - Caption images - Generate music - Generate speech - Fine tune models - Restore images ## Run Open-Source Models Our community has already published thousands of models that are ready to use in production. You can run these with one line of code. ```python import replicate output = replicate.run( "black-forest-labs/flux-schnell:f2ab8a5bfe79f02f0789a146cf5e73d2a4ff2684a98c", input={ "prompt": "An astronaut riding a rainbow unicorn, cinematic, dramatic", "num_outputs": 1, "aspect_ratio": "1:1", "output_format": "webp", "output_quality": 90 } ) print(output) ``` ## Fine-tune Models with Your Own Data You can improve open-source models with your own data to create new models that are better suited to specific tasks. Image models like Flux can generate images of a particular person, object, or style. Train a model: ```python import replicate training = replicate.trainings.create( version="ostris/flux-dev-lora-trainer:1296f0ab2d695af5a1b5eeee6e8ec043145b", input={ "input_images": "https://my-domain/my-input-images.zip", }, destination="electricdreams/flux-fine-tuned" ) print(training) ``` ## Deploy Custom Models You aren't limited to the models on Replicate: you can deploy your own custom models using Cog, our open-source tool for packaging machine learning models. Define your environment in `cog.yaml`: ```yaml build: gpu: true system_packages: - "libgl1-mesa-glx" - "libglib2.0-0" python_version: "3.10" python_packages: - "torch==1.13.1" predict: "predict.py:Predictor" ``` Define predictions in `predict.py`: ```python from cog import BasePredictor, Input, Path import torch class Predictor(BasePredictor): def setup(self): """Load the model in