INNER CODE UNIT · TypeScript

loadTensorflowModel

margelo/react-native-fast-tflite · src/loadTensorflowModel.ts:28

export async function loadTensorflowModel(
  source: ModelSource,
  delegates: TensorflowModelDelegate[]
): Promise<TfliteModel> {
  let uri: string
  if (typeof source === 'number') {
    console.log(`Loading Tensorflow Lite Model ${source}`)
    const asset = Image.resolveAssetSource(source)
    uri = asset.uri
    console.log(`Resolved Model path: ${asset.uri}`)
  } else if (typeof source === 'object' && 'url' in source) {
    uri = source.url
  } else {
    throw new Error(
      'TFLite: Invalid source passed! Source should be either a React Native require(..) or a `{ url: string }` object!'
    )
  }
  const data = await assetLoader.loadAsset(uri)

View source record →