Update app.py

app.py

@@ -12,7 +12,7 @@ def get_current_time():
     current_time = now2.strftime("%Y-%m-%d %H:%M:%S")
     ki = f'{kii} {current_time}'
     return ki
-
+    
 def load_fn(models):
     global models_load
     models_load = {}
@@ -24,10 +24,11 @@ def load_fn(models):
                 m = gr.Interface(lambda txt: None, ['text'], ['image'])
             models_load.update({model: m})
 
-load_fn(models)
-
-num_models = len(models)
-default_models = models[:num_models]
+def load_test(mod):
+    load_fn(mod)
+    num_models = len(mod)
+    default_models = models[:num_models]
+load_test(models_test[1])
 
 def extend_choices(choices):
     return choices + (num_models - len(choices)) * ['NA']