From fa2cc20fc2fb05d5db6d183dd2c1b49d855c5c4b Mon Sep 17 00:00:00 2001 From: Kerry Jiang Date: Sat, 3 Aug 2019 16:55:46 -0700 Subject: [PATCH] removed the direct console output for now, will add logging later --- src/TensorFlowHub/Utils.cs | 23 +++++++---------------- 1 file changed, 7 insertions(+), 16 deletions(-) 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);