Skip to content

Commit 9375f84

Browse files
committed
refactor method
1 parent 6c12f4a commit 9375f84

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/main/java/io/bioimage/modelrunner/tensorflow/v2/api050/Tensorflow2Interface.java

+2-2
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@
2929
import io.bioimage.modelrunner.apposed.appose.Service.TaskStatus;
3030
import io.bioimage.modelrunner.bioimageio.description.ModelDescriptor;
3131
import io.bioimage.modelrunner.bioimageio.description.ModelDescriptorFactory;
32-
import io.bioimage.modelrunner.bioimageio.download.DownloadModel;
32+
import io.bioimage.modelrunner.download.FileDownloader;
3333
import io.bioimage.modelrunner.engine.DeepLearningEngineInterface;
3434
import io.bioimage.modelrunner.engine.EngineInfo;
3535
import io.bioimage.modelrunner.exceptions.LoadModelException;
@@ -242,7 +242,7 @@ private void unzipTfWeights(ModelDescriptor descriptor) throws LoadModelExceptio
242242
System.out.println("Unzipping model...");
243243
ZipUtils.unzipFolder(new File(modelFolder, source).getAbsolutePath(), modelFolder);
244244
} else {
245-
source = DownloadModel.getFileNameFromURLString(source);
245+
source = FileDownloader.getFileNameFromURLString(source);
246246
System.out.println("Unzipping model...");
247247
ZipUtils.unzipFolder(modelFolder + File.separator + source, modelFolder);
248248
}

0 commit comments

Comments
 (0)