replicate-ruby 0.2.0 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 82385c582239598bb4b62a3e133228b7e6fc417ea57a047c1489e5d94d787faf
-  data.tar.gz: 4ccecfdb33f1a4cf3218587c7ed4fa28475da1ca13369e59d5875b07e4f092f8
+  metadata.gz: '0840bf61f6fc1f27234815e501bc2f68a40e1af9b001c5e0df782d0e4083f2d6'
+  data.tar.gz: bc055f8e4f403c8917e47fd90155ae6e4d4c2216a56f7275ee7e8c8d4d2eabd4
 SHA512:
-  metadata.gz: c8edef78b1309f5873997b33b1de5de82f2f8982ce2ec3ed32b0cdeca0e9038c77b4d56e6173441d106160e77fb1e0d2311024d5cae2c68a1fc91ef6c62cb725
-  data.tar.gz: 56958524475fbd6636e1eb3f409b4dad100fe8179a982a746a0ee2ea4356ffa66724269a9da0e226e374bbe34c2526aacf3a358e5847ed55429ebf06700f2383
+  metadata.gz: 47250571e1bcafbe1b77238243229665b97e756d4a3eed92f1bbcc2f99addaa23baef4074b49f7bf784711c4b887b6e3c8c21cc019693e5a79b1b7fe6fea778c
+  data.tar.gz: 6e5bc89eeb116c32a81f246d044e782b85a230df7c4d19e3c3ddfb2ac2a54f5c1dc5a7772f3b2101c4a267a55b20f2a1a9e58d7c0a30fa2d950de290179775ec

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    replicate-ruby (0.1.7)
+    replicate-ruby (0.2.1)
       addressable
       faraday (>= 2.0)
       faraday-multipart

data/README.md

@@ -60,14 +60,13 @@ There is support for the [experimental dreambooth endpoint](https://replicate.co
 
 First, upload your training dataset:
 
-```
-upload = Replicate.client.create_upload
-upload.attach('tmp/data.zip') # replace with the path to your zip file
+```ruby
+upload = Replicate.client.upload_zip('tmp/data.zip') # replace with the path to your zip file
 ```
 
 Then start training a new model using, for instance:
 
-```
+```ruby
 training = Replicate.client.create_training(
   input: {
     instance_prompt: "zwx style",
@@ -81,7 +80,7 @@ training = Replicate.client.create_training(
 
 As soon as the model has finished training, you can run predictions on it:
 
-```
+```ruby
 prediction = Replicate.client.create_prediction(
   input: {
     prompt: 'your prompt, zwx style'

data/lib/replicate/client/upload.rb

@@ -4,10 +4,18 @@ module Replicate
   class Client
     # Methods for the Prediction API
     module Upload
+      # Create upload object and upload zip file
+      def upload_zip(zip_path)
+        filename = zip_path.split('/')[-1]
+        upload = create_upload(filename)
+        upload.attach(zip_path)
+        upload
+      end
+
       # Create an upload
       # @see https://replicate.com/blog/dreambooth-api
-      def create_upload
-        response = dreambooth_endpoint.post("upload/data.zip")
+      def create_upload(filename = 'data.zip')
+        response = dreambooth_endpoint.post("upload/#{filename}")
         Replicate::Record::Upload.new(self, response)
       end
 

data/lib/replicate/version.rb

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

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: replicate-ruby
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.2.1
 platform: ruby
 authors:
 - Dreaming Tulpa