text-to-image

Runtime error

App Files Files Community

ccxccc commited on Mar 10, 2024

Commit

dd80569

verified ·

1 Parent(s): 8f8f343

Update app.py

Browse files

Files changed (1) hide show

app.py +46 -68

app.py CHANGED Viewed

@@ -2,83 +2,61 @@ import gradio as gr
 from random import randint
 from all_models import models
-def load_fn(models):
-    global models_load
-    models_load = {}
-    for model in models:
-        if model not in models_load.keys():
-            try:
-                m = gr.load(f'models/{model}')
-            except Exception as error:
-                m = gr.Interface(lambda txt: None, ['text'], ['image'])
-            models_load.update({model: m})
-load_fn(models)
 num_models = 6
 default_models = models[:num_models]
-def extend_choices(choices):
-    return choices + (num_models - len(choices)) * ['NA']
-def update_imgbox(choices):
-    choices_plus = extend_choices(choices)
-    return [gr.Image(None, label = m, visible = (m != 'NA')) for m in choices_plus]
-def gen_fn(model_str, prompt):
-    if model_str == 'NA':
         return None
-    noise = str('') #str(randint(0, 99999999999))
-    return models_load[model_str](f'{prompt} {noise}')
 with gr.Blocks() as demo:
-    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')
-        stop_button = gr.Button('Stop', variant = 'secondary', interactive = False)
-        gen_button.click(lambda s: gr.update(interactive = True), None, stop_button)
-        gr.HTML(
-        """
-            <div style="text-align: center; max-width: 1200px; margin: 0 auto;">
-              <div>
-                <body>
-                <div class="center"><p style="margin-bottom: 10px; color: #000000;">Scroll down to see more images and select models.</p>
-                </div>
-                </body>
-              </div>
-            </div>
-        """
-               )
         with gr.Row():
-            output = [gr.Image(label = m, min_width=480) for m in default_models]
-            current_models = [gr.Textbox(m, visible = False) for m in default_models]
-            for m, o in zip(current_models, output):
-                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 686 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)
         with gr.Row():
-            gr.HTML(
-    """
-        <div class="footer">
-        <p> Based on the <a href="https://huggingface.co/spaces/derwahnsinn/TestGen">TestGen</a> Space by derwahnsinn, the <a href="https://huggingface.co/spaces/RdnUser77/SpacIO_v1">SpacIO</a> Space by RdnUser77 and Omnibus's Maximum Multiplier!
-        </p>
-    """
-)
-demo.queue(concurrency_count = 200)
-demo.launch()

 from random import randint
 from all_models import models
+def load_models(model_names):
+    loaded_models = {}
+    for model_name in model_names:
+        try:
+            model = gr.load(f'models/{model_name}')
+        except Exception:
+            model = gr.Interface(lambda txt: "Model load error", inputs=["text"], outputs=["image"])
+        loaded_models[model_name] = model
+    return loaded_models
+models_load = load_models(models)
 num_models = 6
 default_models = models[:num_models]
+def extend_choices(choices, num_models=6):
+    return choices + ['NA'] * (num_models - len(choices))
+def update_image_boxes(choices):
+    extended_choices = extend_choices(choices)
+    return [gr.Image(label=model, visible=(model != 'NA')) for model in extended_choices]
+def generate_image(model_name, prompt):
+    if model_name == 'NA':
         return None
+    # If you decide to add randomness to the prompt, uncomment the next line
+    # prompt += f" {randint(0, 999999999)}"
+    return models_load[model_name](prompt)
 with gr.Blocks() as demo:
+    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")
+        stop_button = gr.Button("Stop", variant="secondary")
+        gen_button.click(lambda _: gr.update(interactive=True), None, stop_button)
         with gr.Row():
+            output_images = [gr.Image(label=model, min_width=480) for model in default_models]
+            model_textboxes = [gr.Textbox(model, visible=False) for model in default_models]
+            for model_textbox, output_image in zip(model_textboxes, output_images):
+                gen_event = gen_button.click(generate_image, [model_textbox, txt_input], output_image)
+                stop_button.click(lambda _: gr.update(interactive=False), None, stop_button, cancels=[gen_event])
+        with gr.Accordion("Model selection"):
+            model_choice = gr.CheckboxGroup(choices=models, label=f"Choose up to {num_models} models from the 686 available!", value=default_models, multiselect=True, max_choices=num_models, interactive=True)
+            model_choice.change(update_image_boxes, model_choice, output_images)
+            model_choice.change(lambda choices: extend_choices(choices, num_models), model_choice, model_textboxes)
         with gr.Row():
+            gr.HTML("""
+                <div class="footer">
+                    <p>Based on spaces by derwahnsinn, RdnUser77, and the Omnibus Maximum Multiplier!</p>
+                </div>
+            """)
+demo.queue(concurrency_count=200)
+demo.launch()