Update app.py

app.py

@@ -3,96 +3,75 @@ import os
 import sys
 from pathlib import Path
 
-
 text_gen=gr.Interface.load("spaces/Omnibus/MagicPrompt-Stable-Diffusion")
 
-
-proc1=gr.Interface.load("models/dreamlike-art/dreamlike-photoreal-2.0")
-proc2=gr.Interface.load("models/dreamlike-art/dreamlike-photoreal-2.0")
-proc3=gr.Interface.load("models/dreamlike-art/dreamlike-photoreal-2.0")
-proc4=gr.Interface.load("models/dreamlike-art/dreamlike-photoreal-2.0")
-
-proc5=gr.Interface.load("models/dreamlike-art/dreamlike-diffusion-1.0")
-proc6=gr.Interface.load("models/dreamlike-art/dreamlike-diffusion-1.0")
-proc7=gr.Interface.load("models/dreamlike-art/dreamlike-diffusion-1.0")
-proc8=gr.Interface.load("models/dreamlike-art/dreamlike-diffusion-1.0")
-
-
-def send_it1(inputs,proc1=proc1):
-        output1=proc1(inputs)
+proc1=gr.Interface.load("models/dreamlike-art/dreamlike-photoreal-2.0", preprocess=True, postprocess=False)
+proc2=gr.Interface.load("models/dreamlike-art/dreamlike-photoreal-2.0", preprocess=True, postprocess=False)
+proc3=gr.Interface.load("models/dreamlike-art/dreamlike-photoreal-2.0", preprocess=True, postprocess=False)
+proc4=gr.Interface.load("models/dreamlike-art/dreamlike-photoreal-2.0", preprocess=True, postprocess=False)
+proc5=gr.Interface.load("models/dreamlike-art/dreamlike-diffusion-1.0", preprocess=True, postprocess=False)
+proc6=gr.Interface.load("models/dreamlike-art/dreamlike-diffusion-1.0", preprocess=True, postprocess=False)
+proc7=gr.Interface.load("models/dreamlike-art/dreamlike-diffusion-1.0", preprocess=True, postprocess=False)
+proc8=gr.Interface.load("models/dreamlike-art/dreamlike-diffusion-1.0", preprocess=True, postprocess=False)
+
+def send_it1(inputs, proc=proc1):
+        output1=proc(inputs)
         return(output1)
-def send_it2(inputs,proc2=proc2):
-        output2=proc2(inputs)
+def send_it2(inputs, proc=proc2):
+        output2=proc(inputs)
         return(output2)
-def send_it3(inputs,proc3=proc3):
-        output3=proc3(inputs)
+def send_it3(inputs, proc=proc3):
+        output3=proc(inputs)
         return(output3)
-def send_it4(inputs,proc4=proc4):
-        output4=proc4(inputs)
+def send_it4(inputs, proc=proc4):
+        output4=proc(inputs)
         return(output4)
-
-def send_it5(inputs,proc5=proc5):
-        output5=proc5(inputs)
+def send_it5(inputs, proc=proc5):
+        output5=proc(inputs)
         return(output5)
-def send_it6(inputs,proc6=proc6):
-        output6=proc6(inputs)
+def send_it6(inputs, proc=proc6):
+        output6=proc(inputs)
         return(output6)
-def send_it7(inputs,proc7=proc7):
-        output7=proc7(inputs)
+def send_it7(inputs, proc=proc7):
+        output7=proc(inputs)
         return(output7)
-def send_it8(inputs,proc8=proc8):
-        output8=proc8(inputs)
+def send_it8(inputs, proc=proc8):
+        output8=proc(inputs)
         return(output8)
 
-
 def get_prompts(prompt_text):
     return text_gen(prompt_text)
 
-
-
 with gr.Blocks() as myface:
     with gr.Row():
-            
-            
-            input_text=gr.Textbox(label="Short Prompt")
-            see_prompts=gr.Button("Magic Prompt")
+        input_text=gr.Textbox(label="Short Prompt")
+        see_prompts=gr.Button("Magic Prompt")
     with gr.Row():
+        prompt=gr.Textbox(label="Prettified Prompt")
+        run=gr.Button("Launch")
+
         
-            prompt=gr.Textbox(label="Prettified Prompt")
-            run=gr.Button("Launch")
-        
-    
     with gr.Row():
-            output1=gr.Image(label="DreamlikeArt PhotoReal 1")
-            output2=gr.Image(label="DreamlikeArt PhotoReal 2")
-            output3=gr.Image(label="DreamlikeArt PhotoReal 3")
-            output4=gr.Image(label="DreamlikeArt PhotoReal 4")
+        output1=gr.Image(label="DreamlikeArt PhotoReal 1")
+        output2=gr.Image(label="DreamlikeArt PhotoReal 2")
+        output3=gr.Image(label="DreamlikeArt PhotoReal 3")
+        output4=gr.Image(label="DreamlikeArt PhotoReal 4")
     with gr.Row():
-            output5=gr.Image(label="DreamlikeArt Diffusion 1")
-            output6=gr.Image(label="DreamlikeArt Diffusion 2")
-            output7=gr.Image(label="DreamlikeArt Diffusion 3")
-            output8=gr.Image(label="DreamlikeArt Diffusion 4")        
-
-
-    #def set_models(model_name1, model_name2, model_name3, model_name4):
-
-    #return(proc1,proc2,proc3,proc4)
-
-    #run.click(set_models, inputs=[model_name1, model_name2, model_name3, model_name4], outputs=[proc1,proc2,proc3,proc4])
-    #run.click(send_it, inputs=[prompt], outputs=[output1, output2, output3, output4])
-
-
+        output5=gr.Image(label="DreamlikeArt Diffusion 1")
+        output6=gr.Image(label="DreamlikeArt Diffusion 2")
+        output7=gr.Image(label="DreamlikeArt Diffusion 3")
+        output8=gr.Image(label="DreamlikeArt Diffusion 4")
+    
     see_prompts.click(get_prompts, inputs=[input_text], outputs=[prompt])
-        
-    run.click(send_it1, inputs=[prompt], outputs=[output1])
-    run.click(send_it2, inputs=[prompt], outputs=[output2])
-    run.click(send_it3, inputs=[prompt], outputs=[output3])
-    run.click(send_it4, inputs=[prompt], outputs=[output4])
-    run.click(send_it5, inputs=[prompt], outputs=[output5])
-    run.click(send_it6, inputs=[prompt], outputs=[output6])
-    run.click(send_it7, inputs=[prompt], outputs=[output7])
-    run.click(send_it8, inputs=[prompt], outputs=[output8])
-
+    
+    run.click(send_it1, inputs=[prompt], outputs=[output1], extra_args=[proc1])
+    run.click(send_it2, inputs=[prompt], outputs=[output2], extra_args=[proc2])
+    run.click(send_it3, inputs=[prompt], outputs=[output3], extra_args=[proc3])
+    run.click(send_it4, inputs=[prompt], outputs=[output4], extra_args=[proc4])
+    run.click(send_it5, inputs=[prompt], outputs=[output5], extra_args=[proc5])
+    run.click(send_it6, inputs=[prompt], outputs=[output6], extra_args=[proc6])
+    run.click(send_it7, inputs=[prompt], outputs=[output7], extra_args=[proc7])
+    run.click(send_it8, inputs=[prompt], outputs=[output8], extra_args=[proc8])
 
 myface.queue(concurrency_count=8)
 myface.launch(enable_queue=True, inline=True)