Update app.py

app.py

@@ -62,15 +62,6 @@ async def infer(model_str, prompt, timeout):
         return image
     return None
 
-with gr.Blocks() as demo:
-    gr.HTML(
-    """
-        <div>
-        <p> <center>For simultaneous generations without hidden queue check out <a href="https://huggingface.co/spaces/Yntec/ToyWorld">Toy World</a>! For more options like single model x6 check out <a href="https://huggingface.co/spaces/John6666/Diffusion80XX4sg">Diffusion80XX4sg</a> by John6666!</center>
-        </p></div>
-    """
-)  
-
 def gen_fn(model_str, prompt):
     if model_str == 'NA':
         return None
@@ -118,7 +109,14 @@ CSS="""
 """
 
 with gr.Blocks(theme='Nymbo/Nymbo_Theme', fill_width=True, css=CSS) as demo:
-    with gr.Tab('The Dream'):
+    gr.HTML(
+    """
+        <div>
+        <p> <center>For simultaneous generations without hidden queue check out <a href="https://huggingface.co/spaces/Yntec/ToyWorld">Toy World</a>! For more options like single model x6 check out <a href="https://huggingface.co/spaces/John6666/Diffusion80XX4sg">Diffusion80XX4sg</a> by John6666!</center>
+        </p></div>
+    """
+)  
+    with gr.Tab('Huggingface Diffusion'):
         with gr.Column(scale=2):
             txt_input = gr.Textbox(label='Your prompt:', lines=4)
             with gr.Row():
@@ -154,38 +152,6 @@ with gr.Blocks(theme='Nymbo/Nymbo_Theme', fill_width=True, css=CSS) as demo:
                 #model_choice.change(update_imgbox_gallery, model_choice, output)
                 model_choice.change(extend_choices, model_choice, current_models)
 
-    with gr.Tab('Single model'):
-        with gr.Column(scale=2):
-            model_choice2 = gr.Dropdown(models, label='Choose model', value=models[0])
-            txt_input2 = gr.Textbox(label='Your prompt:', lines=4)
-            
-            max_images = 6
-            num_images = gr.Slider(1, max_images, value=max_images, step=1, label='Number of images')
-            
-            with gr.Row():
-                gen_button2 = gr.Button('Generate', scale=2)
-                stop_button2 = gr.Button('Stop', variant='secondary', interactive=False, scale=1)
-                gen_button2.click(lambda: gr.update(interactive = True), None, stop_button2)
-
-        with gr.Column(scale=1):
-            with gr.Group():
-                with gr.Row():
-                    output2 = [gr.Image(label = '', show_download_button=True, elem_classes="output", interactive=False, min_width=80, visible=True, show_share_button=False, show_label=False) for _ in range(max_images)]
-                    #output2 = [gr.Image(label = '') for _ in range(max_images)]
-
-        with gr.Column(scale=2):
-            gallery2 = gr.Gallery(label="Output", show_download_button=True, elem_classes="gallery",
-                                interactive=False, show_share_button=True, container=True, format="png",
-                                preview=True, object_fit="cover", columns=2, rows=2) 
-
-        for i, o in enumerate(output2):
-            img_i = gr.Number(i, visible = False)
-            num_images.change(lambda i, n: gr.update(visible = (i < n)), [img_i, num_images], o)
-            #gen_event2 = gen_button2.click(lambda i, n, m, t: gen_fn(m, t) if (i < n) else None, [img_i, num_images, model_choice2, txt_input2], o)
-            gen_event2 = gr.on(triggers=[gen_button2.click, txt_input2.submit], fn=lambda i, n, m, t: gen_fn(m, t) if (i < n) else None, inputs=[img_i, num_images, model_choice2, txt_input2], outputs=[o])
-            o.change(add_gallery, [o, model_choice2, gallery2], [gallery2])
-            stop_button2.click(lambda: gr.update(interactive = False), None, stop_button2, cancels = [gen_event2])
-
     gr.Markdown("Based on the [TestGen](https://huggingface.co/spaces/derwahnsinn/TestGen) Space by derwahnsinn, the [SpacIO](https://huggingface.co/spaces/RdnUser77/SpacIO_v1) Space by RdnUser77 and Omnibus's Maximum Multiplier!")
 
 demo.queue()