replicate-api 0.1.2 → 0.1.3
Sign up to get free protection for your applications and to get access to all the features.
- checksums.yaml +4 -4
- data/Gemfile.lock +1 -1
- data/README.md +1 -1
- data/lib/replicate/model.rb +1 -1
- data/lib/replicate/version.rb +1 -1
- metadata +1 -1
checksums.yaml
CHANGED
@@ -1,7 +1,7 @@
|
|
1
1
|
---
|
2
2
|
SHA256:
|
3
|
-
metadata.gz:
|
4
|
-
data.tar.gz:
|
3
|
+
metadata.gz: a596e76f9ab64a522810b88fa4d816f4e3e465938ae9601a9d349c9e4d35bfef
|
4
|
+
data.tar.gz: 191eca1fd5545dea9127146d36f0aabe8c2ebf552a5647c2a4e43791f46729b7
|
5
5
|
SHA512:
|
6
|
-
metadata.gz:
|
7
|
-
data.tar.gz:
|
6
|
+
metadata.gz: 93e354f2cd6dcf5897d5222a0438c04c726a3c0e0d9eb9b6f83c5a5e543b3fe2ce04ca78470a3469de9369f2e5d2b3747b7007cf7aa5107dd26ae03ef3800ee9
|
7
|
+
data.tar.gz: 58b2128c462a265cc7a65ec98b7ece9c43d6e25b354a7b39a9bd256a4b043e5884b6ec600b4fbb281d9c4bccaca18aa09c72a5b7c40fd9aab7b468a9f13215be
|
data/Gemfile.lock
CHANGED
data/README.md
CHANGED
@@ -25,7 +25,7 @@ Or install it yourself as:
|
|
25
25
|
You can run a model and get its output:
|
26
26
|
|
27
27
|
```ruby
|
28
|
-
replicate = Replicate::Client.new(
|
28
|
+
replicate = Replicate::Client.new(token: "#{YOUR API KEY HERE}")
|
29
29
|
model = replicate.models("stability-ai/stable-diffusion")
|
30
30
|
model.predict(prompt="a 19th century portrait of a wombat gentleman")
|
31
31
|
```
|
data/lib/replicate/model.rb
CHANGED
@@ -9,7 +9,7 @@ module Replicate
|
|
9
9
|
@version_id = response.body["latest_version"]["id"]
|
10
10
|
end
|
11
11
|
|
12
|
-
def predict(prompt
|
12
|
+
def predict(prompt, height=512, width=512)
|
13
13
|
body = {version: @version_id, input: {prompt: prompt, height: height, width: width}}
|
14
14
|
response = @client.requests(method: "POST", path: "predictions", **body)
|
15
15
|
poll(prediction_id: response.body["id"])
|
data/lib/replicate/version.rb
CHANGED