Update app.py

app.py

@@ -4,6 +4,7 @@ import sys
 from pathlib import Path
 
 models = [
+    {"name": "Best Model", "url": "dreamlike-art/dreamlike-photoreal-2.0"},
     {"name": "Claudfuen 1", "url": "claudfuen/photorealistic-fuen-v1"},
     {"name": "Deliberate", "url": "Masagin/Deliberate"},
     {"name": "Dreamlike Photoreal", "url": "dreamlike-art/dreamlike-photoreal-2.0"},
@@ -14,7 +15,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:
@@ -45,7 +46,7 @@ with gr.Blocks() as myface:
 
     with gr.Row():
         with gr.Row():
-            input_text = gr.Textbox(label="Prompt idea", lines=1)
+            input_text = gr.Textbox(label="Prompt idea", placeholder="Eg. Eyewear model", lines=1)
             # Model selection dropdown
             model_name1 = gr.Dropdown(
                 label="Choose Model",
@@ -61,29 +62,29 @@ with gr.Blocks() as myface:
     with gr.Row():
         output1 = gr.Image(label="")
         output2 = gr.Image(label="")
-    with gr.Row():
-        magic1 = gr.Textbox(lines=2)
-        magic2 = gr.Textbox(lines=2)
-    with gr.Row():            
         output3 = gr.Image(label="")
-        output4 = gr.Image(label="")
     with gr.Row():
-        magic3 = gr.Textbox(lines=2)
-        magic4 = gr.Textbox(lines=2)
+        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():
-        magic5 = gr.Textbox(lines=2)
-        magic6 = gr.Textbox(lines=2)
-    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)
+    with gr.Row():
         output7 = gr.Image(label="")
         output8 = gr.Image(label="")
+        output9 = gr.Image(label="")
     with gr.Row():
-        magic7 = gr.Textbox(lines=2)
-        magic8 = gr.Textbox(lines=2)
+        magic7 = gr.Textbox(label="Generated Prompt", lines=2)
+        magic8 = gr.Textbox(label="Generated Prompt", lines=2)
+        magic9 = 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, output7, output8, output9])
 
     run.click(send_it, inputs=[magic1, model_name1], outputs=[output1])
     run.click(send_it, inputs=[magic2, model_name1], outputs=[output2])
@@ -93,6 +94,7 @@ with gr.Blocks() as myface:
     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])
+    run.click(send_it, inputs=[magic9, model_name1], outputs=[output9])
 
     see_prompts.click(text_it, inputs=[input_text], outputs=[magic1])
     see_prompts.click(text_it, inputs=[input_text], outputs=[magic2])
@@ -102,6 +104,7 @@ with gr.Blocks() as myface:
     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])
+    see_prompts.click(text_it, inputs=[input_text], outputs=[magic9])
 
 myface.queue(concurrency_count=200)
 myface.launch(inline=True, show_api=False, max_threads=400)