Update app.py

app.py

@@ -67,7 +67,7 @@ def rename_files_with_six_digit_code(file_path):
     return None
 
 
-def download_files_from_huggingface(repo_id, repo_type, max_retries=3):
+def download_files_from_huggingface(repo_id, repo_type, max_retries=10):
     file_list = list_repo_files(repo_id=repo_id, repo_type=repo_type)
     for file in file_list:
         relative_path = os.path.dirname(file)
@@ -91,12 +91,18 @@ def download_files_from_huggingface(repo_id, repo_type, max_retries=3):
                 break
             except (requests.exceptions.ReadTimeout, requests.exceptions.ConnectionError, LocalEntryNotFoundError) as e:
                 if attempt < max_retries - 1:
-                    wait_time = (attempt + 1)*200
+                    wait_time = (attempt + 1)*40
                     print(f"Download failed, retrying in {wait_time} seconds")
                     time.sleep(wait_time)
             except FileExistsError:
                 print(f"{file} Exists")
                 break
+            except Exception as e:
+                if attempt < max_retries - 1:
+                    wait_time = (attempt + 1)*100
+                    print(f"Unknown exception, retrying in {wait_time} seconds")
+                    time.sleep(wait_time)
+                
 
 download_files_from_huggingface(repo_id, 'model')