Function banana_rust_sdk::run

source ·
pub async fn run(
    api_key: &str,
    model_key: &str,
    model_inputs: Value
) -> Result<BananaResponse, BananaError>
Expand description

The main function for calling your model on Banana

Example

use banana_rust_sdk;
use serde::Serialize;

#[tokio::main]
async fn main() {
   #[derive(Serialize)]
   struct ModelInputs {
       prompt: String
   }
   
   let api_key = "API_KEY";
   let model_key = "MODEL_KEY";
   let model_inputs = ModelInputs {
       prompt: "try to predict the next [MASK] of this sentence.".to_string()
   };

   let model_inputs = serde_json::to_value(model_inputs).unwrap();

   let res = banana_rust_sdk::run(api_key, model_key, model_inputs).await.unwrap();
   let json = serde_json::to_value(res).unwrap();
   println!("{:?}", json);
}