This gem bundles the replicate-ruby gem and adds webhook support for replicate.com.
Add this line to your application's Gemfile:
gem 'replicate-rails', require: 'replicate_rails'Setup an initializer and webhook adapter class:
# config/initializers/replicate.rb Replicate.client.api_token = "your-api-token" ReplicateRails.configure do |config| config.webhook_adapter = ReplicateWebhook.new end class ReplicateWebhook def call(prediction) # do your thing end endand mount the route:
# config/routes.rb mount ReplicateRails::Engine => "/replicate/webhook"Now you can run predictions as follows:
model = Replicate.client.retrieve_model("stability-ai/stable-diffusion") version = model.latest_version version.predict(prompt: "a beautiful sunset", "https://yourdomain.tld/replicate/webhook") After checking out the repo, run bin/setup to install dependencies. Then, run rake test to run the tests. You can also run bin/console for an interactive prompt that will allow you to experiment.