Update app.py

app.py

@@ -2,6 +2,13 @@ import gradio as gr
 from random import randint
 from all_models import models
 
+from externalmod import gr_Interface_load
+
+import asyncio
+import os
+from threading import RLock
+lock = RLock()
+HF_TOKEN = os.environ.get("HF_TOKEN") if os.environ.get("HF_TOKEN") else None # If private or gated models aren't used, ENV setting is unnecessary.
 
 
 def load_fn(models):
@@ -11,9 +18,10 @@ def load_fn(models):
     for model in models:
         if model not in models_load.keys():
             try:
-                m = gr.load(f'models/{model}')
+                m = gr_Interface_load(f'models/{model}', hf_token=HF_TOKEN)
             except Exception as error:
-                m = gr.Interface(lambda txt: None, ['text'], ['image'])
+                print(error)
+                m = gr.Interface(lambda: None, ['text'], ['image'])
             models_load.update({model: m})
 
 
@@ -22,17 +30,16 @@ load_fn(models)
 
 num_models = 2
 default_models = models[:num_models]
-
+inference_timeout = 600
 
 
 def extend_choices(choices):
-    return choices + (num_models - len(choices)) * ['NA']
+    return choices[:num_models] + (num_models - len(choices[:num_models])) * ['NA']
 
 
 def update_imgbox(choices):
-    choices_plus = extend_choices(choices)
-    return [gr.Image(None, label = m, visible = (m != 'NA')) for m in choices_plus]
-
+    choices_plus = extend_choices(choices[:num_models])
+    return [gr.Image(None, label=m, visible=(m!='NA')) for m in choices_plus]
     
 def gen_fn(model_str, prompt):
     if model_str == 'NA':