diff --git a/src/TensorFlowHub/Utils.cs b/src/TensorFlowHub/Utils.cs index 72ee9430..3245071f 100644 --- a/src/TensorFlowHub/Utils.cs +++ b/src/TensorFlowHub/Utils.cs @@ -28,22 +28,14 @@ namespace Tensorflow.Hub var fileSaveTo = Path.Combine(dirSaveTo, fileName); if (File.Exists(fileSaveTo)) + return; + + Directory.CreateDirectory(dirSaveTo); + + using (var wc = new WebClient()) { - //TODO:maybe you can check file's hashcode and "donglowad.info" to complete file ... - Console.WriteLine($"{fileSaveTo} already exists."); + await wc.DownloadFileTaskAsync(url, fileSaveTo); } - else - { - if (!Directory.Exists(dirSaveTo)) - Directory.CreateDirectory(dirSaveTo); - - using (var wc = new WebClient()) - { - await wc.DownloadFileTaskAsync(url, fileSaveTo); - } - - } - } public static async Task UnzipAsync(this IModelLoader modelLoader, string zipFile, string saveTo) @@ -52,8 +44,7 @@ namespace Tensorflow.Hub if (!Path.IsPathRooted(saveTo)) saveTo = Path.Combine(AppContext.BaseDirectory, saveTo); - if (!Directory.Exists(saveTo)) - Directory.CreateDirectory(saveTo); + Directory.CreateDirectory(saveTo); if (!Path.IsPathRooted(zipFile)) zipFile = Path.Combine(AppContext.BaseDirectory, zipFile);