Merge pull request #22 from saramaxyz/feature/fix_progress

Fix progress callback
saramaxyz · Oct 11, 2023 · 8bb9ddd · 8bb9ddd
2 parents 739014f + ca61473
commit 8bb9ddd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Sources/AeroEdge/Infrastructure/ModelDownloader.swift b/Sources/AeroEdge/Infrastructure/ModelDownloader.swift
@@ -81,7 +81,7 @@ extension ModelDownloader: URLSessionDownloadDelegate {
     let progress = Float(totalBytesWritten) / Float(totalBytesExpectedToWrite)
     downloadProgress[downloadTask] = progress
 
-    guard let modelName = downloadTask.originalRequest?.url?.lastPathComponent else {
+    guard let modelName = downloadTask.originalRequest?.url?.lastPathComponent.replacingOccurrences(of: ".zip", with: "") else {
       return
     }
     delegate?.modelDownloadProgress(forModel: modelName, progress: progress)