Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import numpy as np
 import random
+from tqdm import tqdm
 
 # import spaces #[uncomment to use ZeroGPU]
 from diffusers import DiffusionPipeline
@@ -25,11 +26,6 @@ def init_model(model_repo_id):
     torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
     pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
     return pipe
-    
-loaded_models = {}
-
-for model in available_models:
-    loaded_models[model] = init_model(model)
 
 # @spaces.GPU #[uncomment to use ZeroGPU]
 def infer(
@@ -153,6 +149,13 @@ with gr.Blocks(css=css) as demo:
                 )
 
         gr.Examples(examples=examples, inputs=[prompt])
+
+        
+    loaded_models = {}
+
+    for model in tqdm(available_models):
+        loaded_models[model] = init_model(model)
+        
     gr.on(
         triggers=[run_button.click, prompt.submit],
         fn=infer,