mirror of https://github.com/0xplaygrounds/rig
37 lines
897 B
Rust
37 lines
897 B
Rust
use rig::audio_generation::AudioGenerationModel;
|
|
use rig::providers::hyperbolic;
|
|
use std::env::args;
|
|
use std::fs::File;
|
|
use std::io::Write;
|
|
use std::path::Path;
|
|
|
|
const DEFAULT_PATH: &str = "./output.mp3";
|
|
|
|
#[tokio::main]
|
|
async fn main() {
|
|
let arguments: Vec<String> = args().collect();
|
|
|
|
let path = if arguments.len() > 1 {
|
|
arguments[1].clone()
|
|
} else {
|
|
DEFAULT_PATH.to_string()
|
|
};
|
|
|
|
let path = Path::new(&path);
|
|
let mut file = File::create_new(path).expect("Failed to create file");
|
|
|
|
let hyperbolic = hyperbolic::Client::from_env();
|
|
|
|
let tts = hyperbolic.audio_generation_model("EN");
|
|
|
|
let response = tts
|
|
.audio_generation_request()
|
|
.text("The quick brown fox jumps over the lazy dog")
|
|
.voice("EN-US")
|
|
.send()
|
|
.await
|
|
.expect("Failed to generate image");
|
|
|
|
let _ = file.write(&response.audio);
|
|
}
|