Update app.py

app.py

@@ -19,8 +19,6 @@ inference_timeout = 400
 default_models = models[:num_models]
 MAX_SEED = 2**32-1
 
-load_fn(models)
-
 def gr_Interface_load(
     name: str,
     src: str | None = None,
@@ -78,6 +76,8 @@ def load_fn(models):
                 m = gr.Interface(lambda: None, ['text'], ['image'])
             models_load.update({model: m})
 
+load_fn(models)
+
 def extend_choices(choices):
     return choices[:num_models] + (num_models - len(choices[:num_models])) * ['NA']