replicate-api 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 36de9d636cd090ca0252d22ffd4cb48cca83b97810f4b235e05814caaa0b35e1
-  data.tar.gz: bf08feae2677ae8cedbf103e9ef93df03f824c6c671e2be352a09f39b439b1e0
+  metadata.gz: bdc195548bd20d60b90ad926efc9a045dabf09f932b9a5a41ee693b394cce59c
+  data.tar.gz: 1058816a27e6fcdbe4853df7d2bb4c102ddaee92933e6b00309dbcd4b8a463d2
 SHA512:
-  metadata.gz: b3bd50c4c5c8d6532f72d32385ebf929e6f32fe3cdbaac79e223131362bcff694873cca78ce7897893d215eb18499829ce27bf80d6f45a96961c30568931437e
-  data.tar.gz: 9bcbedddc860ab97208fe7b5ddd0bd9d87c73eb213ec15429325fc802d5f73221a754b3f720e8da3d16a307d656fde249c0851885f2093fde23633fcde5c02da
+  metadata.gz: e35c8e77c74981fb177a8b325ba36658eee0195163166773697fb1d3c6133ccac29af72f14b1891575687d6a1c5040a98ee0e2e8f8a1027cdf5843bdb67757ab
+  data.tar.gz: 0f04b274aee7be57ab8f266cb87c5daacf83ef8581310afca9d4b416d1077e138d2766b9b57cf267f3a72058ad058e36c225f0ddfddf06379a5ec48b33593998

data/lib/replicate/model.rb

@@ -9,8 +9,8 @@ module Replicate
       @version_id = response.body["latest_version"]["id"]
     end
 
-    def predict(prompt:)
-      body = {version: @version_id, input: {prompt: prompt}}
+    def predict(prompt:, height = 512, width = 512)
+      body = {version: @version_id, input: {prompt: prompt, height: height, width: width}}
       response = @client.requests(method: "POST", path: "predictions", **body)
       poll(prediction_id: response.body["id"])
     end

data/lib/replicate/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Replicate
-  VERSION = "0.1.1"
+  VERSION = "0.1.2"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: replicate-api
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - Olivia Li
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2022-11-27 00:00:00.000000000 Z
+date: 2022-12-06 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: faraday