Package ai.djl.spring.configuration

Class DjlAutoConfiguration

java.lang.Object

ai.djl.spring.configuration.DjlAutoConfiguration

@Configuration
@ConditionalOnMissingBean(ai.djl.repository.zoo.ZooModel.class)
@EnableConfigurationProperties(DjlConfigurationProperties.class)
public class DjlAutoConfiguration
extends Object

Constructor Summary

Constructors

Constructor	Description
DjlAutoConfiguration()

Method Summary

Modifier and Type	Method	Description
ai.djl.repository.zoo.ZooModel<?,?>	model()
Supplier<ai.djl.inference.Predictor<?,?>>	predictorProvider(ai.djl.repository.zoo.ZooModel<?,?> model)	Expected to be used with try-with-resources.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

DjlAutoConfiguration

public DjlAutoConfiguration()

Method Detail

model

@Bean
public ai.djl.repository.zoo.ZooModel<?,?> model()
                                                throws ai.djl.MalformedModelException,
                                                       ai.djl.repository.zoo.ModelNotFoundException,
                                                       IOException

Throws:

ai.djl.MalformedModelException

ai.djl.repository.zoo.ModelNotFoundException

IOException

predictorProvider

@Bean
public Supplier<ai.djl.inference.Predictor<?,?>> predictorProvider(ai.djl.repository.zoo.ZooModel<?,?> model)

Expected to be used with try-with-resources. The provided predictor is AutoCloseable.

Parameters:

model - injected configured model

Returns:

provider of the predictor object