Update app.py

app.py

@@ -48,7 +48,7 @@ models = [
 
 current_model = models[0]
 
-text_gen = gr.Interface.load("spaces/Omnibus/MagicPrompt-Stable-Diffusion_link")
+text_gen = gr.Interface.load("spaces/daspartho/prompt-extend") 
 
 models2 = []
 for model in models:
@@ -64,7 +64,7 @@ def text_it(inputs, text_gen=text_gen):
 def set_model(current_model_index):
     global current_model
     current_model = models[current_model_index]
-    return gr.update(label=f"{current_model['name']}")
+    return gr.update(value=f"{current_model['name']}")
 
 
 def send_it(inputs, model_choice):
@@ -72,16 +72,14 @@ def send_it(inputs, model_choice):
     return proc(inputs)
 
 
-css = """"""
-
-with gr.Blocks(css=css) as myface:
+with gr.Blocks() as myface:
     gr.HTML(
 
     )
 
     with gr.Row():
         with gr.Row():
-            input_text = gr.Textbox(label="Prompt idea", lines=1)
+            input_text = gr.Textbox(label="Prompt idea",  placeholder="", lines=1)
             # Model selection dropdown
             model_name1 = gr.Dropdown(
                 label="Choose Model",
@@ -93,30 +91,25 @@ with gr.Blocks(css=css) as myface:
         with gr.Row():
             see_prompts = gr.Button("Generate Prompts")
             run = gr.Button("Generate Images", variant="primary")
-    with gr.Tab("Main"):
-        with gr.Row():
-            output1 = gr.Image(label=f"{current_model['name']}")
-            output2 = gr.Image(label=f"{current_model['name']}")
-            output3 = gr.Image(label=f"{current_model['name']}")
-            output4 = gr.Image(label=f"{current_model['name']}")
-        with gr.Row():
-            magic1 = gr.Textbox(lines=4)
-            magic2 = gr.Textbox(lines=4)
-            magic3 = gr.Textbox(lines=4)
-            magic4 = gr.Textbox(lines=4)
-
-        with gr.Row():
-            output5 = gr.Image(label=f"{current_model['name']}")
-            output6 = gr.Image(label=f"{current_model['name']}")
-            output7 = gr.Image(label=f"{current_model['name']}")
-            output8 = gr.Image(label=f"{current_model['name']}")
-        with gr.Row():
-            magic5 = gr.Textbox(lines=4)
-            magic6 = gr.Textbox(lines=4)
-            magic7 = gr.Textbox(lines=4)
-            magic8 = gr.Textbox(lines=4)
+    
+    with gr.Row():
+        output1 = gr.Image(label="")
+        output2 = gr.Image(label="")
+        output3 = gr.Image(label="")
+    with gr.Row():
+        magic1 = gr.Textbox(label="Generated Prompt", lines=2)
+        magic2 = gr.Textbox(label="Generated Prompt", lines=2)
+        magic3 = gr.Textbox(label="Generated Prompt", lines=2)
+    with gr.Row():
+        output4 = gr.Image(label="")
+        output5 = gr.Image(label="")
+        output6 = gr.Image(label="")
+    with gr.Row():
+        magic4 = gr.Textbox(label="Generated Prompt", lines=2)
+        magic5 = gr.Textbox(label="Generated Prompt", lines=2)
+        magic6 = gr.Textbox(label="Generated Prompt", lines=2)
 
-    model_name1.change(set_model, inputs=model_name1, outputs=[output1, output2, output3, output4, output5, output6, output7, output8])
+    model_name1.change(set_model, inputs=model_name1, outputs=[output1, output2, output3, output4, output5, output6])
 
     run.click(send_it, inputs=[magic1, model_name1], outputs=[output1])
     run.click(send_it, inputs=[magic2, model_name1], outputs=[output2])
@@ -124,8 +117,6 @@ with gr.Blocks(css=css) as myface:
     run.click(send_it, inputs=[magic4, model_name1], outputs=[output4])
     run.click(send_it, inputs=[magic5, model_name1], outputs=[output5])
     run.click(send_it, inputs=[magic6, model_name1], outputs=[output6])
-    run.click(send_it, inputs=[magic7, model_name1], outputs=[output7])
-    run.click(send_it, inputs=[magic8, model_name1], outputs=[output8])
 
     see_prompts.click(text_it, inputs=[input_text], outputs=[magic1])
     see_prompts.click(text_it, inputs=[input_text], outputs=[magic2])
@@ -133,8 +124,6 @@ with gr.Blocks(css=css) as myface:
     see_prompts.click(text_it, inputs=[input_text], outputs=[magic4])
     see_prompts.click(text_it, inputs=[input_text], outputs=[magic5])
     see_prompts.click(text_it, inputs=[input_text], outputs=[magic6])
-    see_prompts.click(text_it, inputs=[input_text], outputs=[magic7])
-    see_prompts.click(text_it, inputs=[input_text], outputs=[magic8])
 
 myface.queue(concurrency_count=200)
-myface.launch(inline=True, show_api=False, max_threads=400)  
+myface.launch(inline=True, show_api=False, max_threads=400)