replicate-api 0.2.0 → 0.2.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 180c9dd2c75f2335a772aedb8064e410c11afd4aba23b16e85c90b9ea415ff49
-  data.tar.gz: 4a2feaf5e3e75f01588b9f1952b847653968358a80ea5c5ddf860c113e10a28f
+  metadata.gz: b6d16e84d1360f1e1c2fa7988cdefd8275557e5c91f4653863387d8d79391d8c
+  data.tar.gz: 6bb146750f37e31bbef4a744bc95f8069ca5311945aa828c84219e771595e369
 SHA512:
-  metadata.gz: 50b6f576d2c4c8a72df80278c327014da33eda9cb954873e45730fafb809c8b10fd1544123f08ef8a8b0c257f366d94b3006d641f9845862e3972e3c66e087bb
-  data.tar.gz: ad58c277a43eb85171ee303daae1c8142a5543a4071c18c1c76120f4509f2fd5f38166392efc61ed3b70c97a36fa2efe5ab6e5b067ff235f6105c96d302a1c3b
+  metadata.gz: e3f5971aaa49e11f34c9d8008b200ce1988c160994e77ba93c3e578a2d7b7eaaa8785bf94a524611dede4f07c8ad8b2a9b68813ef009ee9591349706a11802f7
+  data.tar.gz: dd77e11d9604246e8cf614af84e0da43e5f92aaf89eb10f701fc210c736fbd7abe23c1e0e17bf62684ba4525ee47885e0487e32c4b383dbf0ce0522ca896d76d

data/lib/replicate/model.rb

@@ -10,7 +10,7 @@ module Replicate
     end
 
     def predict(prompt: "", height: 512, width: 512, image: "")
-      body = {version: @version_id, input: {prompt: prompt, height: height, width: width}}
+      body = {version: @version_id, input: {prompt: prompt, height: height, width: width, image: image}}
       response = @client.requests(method: "POST", path: "predictions", **body)
       poll(prediction_id: response.body["id"])
     end
@@ -22,7 +22,7 @@ module Replicate
         sleep(2)
         poll(prediction_id: prediction_id)
       elsif response.body["status"] == "succeeded" 
-        return response.body["output"][0]
+        return response.body["output"]
       else
         raise response.body["error"]
       end

data/lib/replicate/version.rb

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

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: replicate-api
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.2.2
 platform: ruby
 authors:
 - Olivia Li