Introducing Google AI Edge Portal: Benchmark Edge AI at scale. Sign-up to request access during private preview.

ModelSelector

class ModelSelector

ModelSelector allows to dynamically select a ModelProvider from a given set of providers.

Summary

Public constructors
`ModelSelector(vararg modelProviders: ModelProvider)`
`ModelSelector(modelProviders: Set<ModelProvider>)`

Public functions
`suspend ModelProvider`	`selectModel(env: Environment)` Selects a `ModelProvider` from the given set based on the availability of model files and accelerators, in the order of NPU, GPU, CPU.
`ListenableFuture<ModelProvider>`	`selectModelFuture(env: Environment, lifecycleOwner: LifecycleOwner)` Returns a future that completes when the model is selected and ready to use.

ModelSelector(vararg modelProviders: ModelProvider)

ModelSelector(modelProviders: Set<ModelProvider>)

suspend fun selectModel(env: Environment): ModelProvider

Selects a ModelProvider from the given set based on the availability of model files and accelerators, in the order of NPU, GPU, CPU.

Returns
`ModelProvider`	the selected `ModelProvider`, which is guaranteed to be ready to use.

Throws
`java.lang.IllegalStateException`	if no model is available.

fun selectModelFuture(env: Environment, lifecycleOwner: LifecycleOwner): ListenableFuture<ModelProvider>

Returns a future that completes when the model is selected and ready to use.

NOTE: Kotlin callers should use selectModel instead.