load all image models

app.py

@@ -7,7 +7,7 @@ from serve.leaderboard import build_leaderboard_tab
 from model.model_manager import ModelManager
 from pathlib import Path
 from serve.constants import SERVER_PORT, ROOT_PATH, ELO_RESULTS_DIR
-from model.pre_download import pre_download_all_models, pre_download_video_models
+from model.pre_download import pre_download_all_image_models
 
 def build_combine_demo(models, elo_results_file, leaderboard_table_file):
 
@@ -99,7 +99,7 @@ if __name__ == "__main__":
     elo_results_dir = ELO_RESULTS_DIR
     models = ModelManager()
 
-    #print("Error downloading models:", pre_download_video_models())
+    pre_download_all_image_models()
     
     elo_results_file, leaderboard_table_file = load_elo_results(elo_results_dir)
     demo = build_combine_demo(models, elo_results_file, leaderboard_table_file)

model/pre_download.py

@@ -1,14 +1,14 @@
 from .models import IMAGE_GENERATION_MODELS, IMAGE_EDITION_MODELS, VIDEO_GENERATION_MODELS
 
-def pre_download_all_models():
+def pre_download_all_image_models():
     """
-    Pre-download all models to avoid download delay during the first user request
+    Pre-download all image models to avoid download delay during the first user request
     """
     imagen_dl_error = pre_download_image_models()
     imagedit_dl_error = pre_download_image_models()
-    videogen_dl_error = pre_download_video_models()
+    #videogen_dl_error = pre_download_video_models()
     print("All models downloaded.")
-    print("Models that encountered download error:", "Image Generation:", imagen_dl_error, "Image Edition:", imagedit_dl_error, "Video Generation:", videogen_dl_error)
+    print("Models that encountered download error:", "Image Generation:", imagen_dl_error, "Image Edition:", imagedit_dl_error)#, "Video Generation:", videogen_dl_error)
 
 def pre_download_image_models():
     """