only load video 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
+from model.pre_download import pre_download_all_models, pre_download_video_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()
 
-    pre_download_all_models()
+    print("Error downloading models:", pre_download_video_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

@@ -19,10 +19,10 @@ def pre_download_image_models():
     for model_string in IMAGE_GENERATION_MODELS:
         
         model_lib, model_name, model_type = model_string.split("_")
-        print("Loading image generation model:", model_name)
-
+        
         if model_lib == "imagenhub":
             try:
+                print("Loading image generation model:", model_name)
                 temp_model = imagen_hub.get_model(model_name) # Forcing model to download weight files
                 del temp_model
             except Exception as e:
@@ -42,10 +42,10 @@ def pre_download_image_models():
     for model_string in IMAGE_EDITION_MODELS:
         
         model_lib, model_name, model_type = model_string.split("_")
-        print("Loading image edition model:", model_name)
 
         if model_lib == "imagenhub":
             try:
+                print("Loading image edition model:", model_name)
                 temp_model = imagen_hub.get_model(model_name) # Forcing model to download weight files
                 del temp_model
             except Exception as e:
@@ -65,10 +65,10 @@ def pre_download_video_models():
     for model_string in VIDEO_GENERATION_MODELS:
         
         model_lib, model_name, model_type = model_string.split("_")
-        print("Loading video generation model:", model_name)
 
         if model_lib == "videogenhub":
             try:
+                print("Loading video generation model:", model_name)
                 temp_model = videogen_hub.get_model(model_name) # Forcing model to download weight files
                 del temp_model
             except Exception as e: