fix downloading issue

TNTwise · Aug 30, 2024 · ed2733a · ed2733a
1 parent 6eef9e8
commit ed2733a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/src/DownloadModels.py b/src/DownloadModels.py
@@ -40,6 +40,7 @@ def downloadModel(self, modelFile: str = None, downloadModelPath: str = None):
         )
         title = "Downloading: " + modelFile
         DownloadProgressPopup(link=url, title=title, downloadLocation=downloadModelPath)
+        print("Done")
         if "tar.gz" in self.downloadModelFile:
             print("Extracting File")
             extractTarGZ(self.downloadModelPath)

diff --git a/src/ui/QTcustom.py b/src/ui/QTcustom.py
@@ -157,7 +157,7 @@ def run(self):
 
         totalSize = 0
         with open(self.downloadLocation, "wb") as f:
-            chunk_size = 128
+            chunk_size = 1024
             for chunk in response.iter_content(chunk_size=chunk_size):
                 f.write(chunk)
                 totalSize += chunk_size