From a41a62fb20372530b86e26af5c0939ebaab4ebf4 Mon Sep 17 00:00:00 2001 From: Julia Beliaeva Date: Mon, 1 Aug 2022 20:06:44 +0200 Subject: [PATCH] Use model storage url from the ModelHub property --- .../kotlinx/dl/api/inference/keras/loaders/TFModelHub.kt | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/api/src/main/kotlin/org/jetbrains/kotlinx/dl/api/inference/keras/loaders/TFModelHub.kt b/api/src/main/kotlin/org/jetbrains/kotlinx/dl/api/inference/keras/loaders/TFModelHub.kt index 56b7fec66..ee5554377 100644 --- a/api/src/main/kotlin/org/jetbrains/kotlinx/dl/api/inference/keras/loaders/TFModelHub.kt +++ b/api/src/main/kotlin/org/jetbrains/kotlinx/dl/api/inference/keras/loaders/TFModelHub.kt @@ -225,7 +225,7 @@ public class TFModelHub(cacheDirectory: File) : ModelHub(cacheDirectory) { if (noTop) modelDirectory += "/notop" val relativeConfigPath = modelDirectory + MODEL_CONFIG_FILE_NAME - val configURL = AWS_S3_URL + modelDirectory + MODEL_CONFIG_FILE_NAME + val configURL = awsS3Url + modelDirectory + MODEL_CONFIG_FILE_NAME val dir = File(cacheDirectory.absolutePath + modelDirectory) if (!dir.exists()) Files.createDirectories(dir.toPath()) @@ -249,7 +249,7 @@ public class TFModelHub(cacheDirectory: File) : ModelHub(cacheDirectory) { if (noTop) modelDirectory += "/notop" val relativeWeightsPath = modelDirectory + WEIGHTS_FILE_NAME - val weightsURL = AWS_S3_URL + modelDirectory + WEIGHTS_FILE_NAME + val weightsURL = awsS3Url + modelDirectory + WEIGHTS_FILE_NAME val fileName = cacheDirectory.absolutePath + relativeWeightsPath val file = File(fileName)