![Latest Version] ![Documentation] ![License]
This is an Unofficial Rust client for Replicate.
sh
cargo add replicate_rust
Grab your token from replicate.com/account and set it as an environment variable:
export REPLICATE_API_TOKEN=<your token>
```rust use replicate_rust::Replicate;
let replicate = Replicate::new();
// Construct the inputs. let mut inputs = std::collections::HashMap::new(); inputs.insert("prompt", "a 19th century portrait of a wombat gentleman");
let version = String::from("stability-ai/stable-diffusion:27b93a2413e7f36cd83da926f3656280b2931564ff050bf9575f1fdf9bcd7478");
// Run the model. let result = replicate.run(version, inputs);
// Print the result. match result { Ok(result) => println!("Success : {:?}", result.output), Err(e) => println!("Error : {}", e), } // Some(Array [String("https://pbxt.replicate.delivery/QLDGe2rXuIQ9ByMViQEXrYCkKfDi9I3YWAzPwWsDZWMXeN7iA/out-0.png")]) ```
You can start a model and run it in the background:
```rust let replicate = Replicate::new();
// Construct the inputs. let mut inputs = std::collections::HashMap::new(); inputs.insert("prompt", "a 19th century portrait of a wombat gentleman");
let version = String::from("stability-ai/stable-diffusion:27b93a2413e7f36cd83da926f3656280b2931564ff050bf9575f1fdf9bcd7478");
// Run the model. let mut prediction = replicate.predictions.create(version, inputs);
println!("{}", prediction.status) // 'starting'
prediction.reload() println!("{}", prediction.status) // 'processing'
println!("{}", prediction.logs) // Some("Using seed: 3599 // 0%| | 0/50 [00:00, ?it/s] // 4%|▍ | 2/50 [00:00<00:04, 10.00it/s] // 8%|▊ | 4/50 [00:00<00:03, 11.56it/s] // ")
prediction.wait()
println!("{}", prediction.status) // 'succeeded'
match prediction.wait() { Ok(result) => println!("Success : {:?}", result.output), Err(e) => println!("Error : {}", e), } // Success : Some(Array [String("https://pbxt.replicate.delivery/QLDGe2rXuIQ9ByMViQEXrYCkKfDi9I3YWAzPwWsDZWMXeN7iA/out-0.png")]) ```
You can cancel a running prediction:
```rust
let replicate = Replicate::new();
// Construct the inputs. let mut inputs = std::collections::HashMap::new(); inputs.insert("prompt", "a 19th century portrait of a wombat gentleman");
let version = String::from("stability-ai/stable-diffusion:27b93a2413e7f36cd83da926f3656280b2931564ff050bf9575f1fdf9bcd7478");
// Run the model. let mut prediction = replicate.predictions.create(version, inputs);
println!("{}", prediction.status) // 'starting'
prediction.cancel()
prediction.reload()
println!("{}", prediction.status) // 'cancelled' ```
You can list all the predictions you've run:
rust
match replicate.predictions.list() {
Ok(result) => println!("Success : {:?}", result),
Err(e) => println!("Error : {}", e),
}
// Success : ListPredictions { ... }
```rust match replicate.models.get(String::from("replicate"), String::from("hello-world")) { Ok(result) => println!("Success : {:?}", result), Err(e) => println!("Error : {}", e), };
// Success : GetModel { ... } ```
```rust match replicate .models .versions .list(String::from("replicate"), String::from("hello-world")) { Ok(result) => println!("Success : {:?}", result), Err(e) => println!("Error : {}", e), }; // Success : ListModelVersions { ... } ``````
rust
match replicate.models.versions.get(
String::from("kvfrans"),
String::from("clipdraw"),
String::from("5797a99edc939ea0e9242d5e8c9cb3bc7d125b1eac21bda852e5cb79ede2cd9b"),
) {
Ok(result) => println!("Success : {:?}", result),
Err(e) => println!("Error : {}", e),
}
// Success : GetModelVersion { ... }
```rust match replicate.collection.get(String::from("audio-generation")) { Ok(result) => println!("Success : {:?}", result), Err(e) => println!("Error : {}", e), }
// Success : GetCollectionModels { ... } ```
```rust match replicate.collection.list() { Ok(result) => println!("Success : {:?}", result), Err(e) => println!("Error : {}", e), }
// Success : ListCollectionModels { ... } ```