Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from random import randint
 from all_models import models
-
+from externalmod import gr_Interface_load
 
 
 def load_fn(models):
@@ -11,7 +11,7 @@ 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}')
             except Exception as error:
                 m = gr.Interface(lambda txt: None, ['text'], ['image'])
             models_load.update({model: m})
@@ -26,11 +26,11 @@ default_models = models[:num_models]
 
 
 def extend_choices(choices):
-    return choices + (num_models - len(choices)) * ['NA']
+    return choices[0:num_models] + (num_models - len(choices[0:num_models])) * ['NA']
 
 
 def update_imgbox(choices):
-    choices_plus = extend_choices(choices)
+    choices_plus = extend_choices(choices[0:num_models])
     return [gr.Image(None, label = m, visible = (m != 'NA')) for m in choices_plus]
 
     
@@ -43,9 +43,17 @@ def gen_fn(model_str, prompt):
 
 
 with gr.Blocks() as demo:
+    gr.HTML(
+    """
+        <div>
+        <p> <center>For simultaneous generations without queue check out <a href="https://huggingface.co/spaces/Yntec/ToyWorld">Toy World</a>!</center>
+        </p></div>
+    """
+)  
+    
     with gr.Tab('The Dream'): 
-        txt_input = gr.Textbox(label = 'Your prompt:', lines=4).style(container=False,min_width=1200)
-        gen_button = gr.Button('Generate up to 6 images in up to 3 minutes total')
+        txt_input = gr.Textbox(label = 'Your prompt:', lines=4) #.style(container=False,min_width=1200)
+        gen_button = gr.Button('Generate up to 6 images from 1 minute to 18 minutes total wait time')
         stop_button = gr.Button('Stop', variant = 'secondary', interactive = False)
         gen_button.click(lambda s: gr.update(interactive = True), None, stop_button)
         gr.HTML(
@@ -68,7 +76,11 @@ with gr.Blocks() as demo:
                 gen_event = gen_button.click(gen_fn, [m, txt_input], o)
                 stop_button.click(lambda s: gr.update(interactive = False), None, stop_button, cancels = [gen_event])
         with gr.Accordion('Model selection'):
-            model_choice = gr.CheckboxGroup(models, label = f'Choose up to {num_models} different models from the 866 available!', value = default_models, multiselect = True, max_choices = num_models, interactive = True, filterable = False)
+            #model_choice = gr.CheckboxGroup(models, label = f'Choose up to {num_models} different models from the 866 available!', value = default_models, multiselect = True, max_choices = num_models, interactive = True, filterable = False)
+            #model_choice.change(update_imgbox, model_choice, output)
+            #model_choice.change(extend_choices, model_choice, current_models)
+            model_choice = gr.CheckboxGroup(models, label = f'Choose up to {num_models} different models from the 874 available!', value = default_models, interactive = True)
+            #model_choice.change(lambda x: x[0:num_models], [model_choice], [model_choice])
             model_choice.change(update_imgbox, model_choice, output)
             model_choice.change(extend_choices, model_choice, current_models)
         with gr.Row():
@@ -80,5 +92,5 @@ with gr.Blocks() as demo:
     """
 )
 
-demo.queue(concurrency_count = 200)                        
+demo.queue()                        
 demo.launch()