Update app.py

app.py

@@ -7,9 +7,19 @@ models = [
 
 current_model = models[0]
 
+models2 = []
+for model in models:
+    model_url = f"models/{model['url']}"
+    loaded_model = gr.Interface.load(model_url, live=True, preprocess=True)
+    models2.append(loaded_model)
+
 def flip_text(x):
     return x[::-1]
 
+def send_it(inputs, model_choice):
+    proc = models2[model_choice]
+    return proc(inputs)
+
 
 def flip_image(x):
     return np.fliplr(x)
@@ -38,6 +48,14 @@ with gr.Blocks() as pan:
          )
          input_text = gr.Textbox(label="Prompt idea",)
          output1 = gr.Image(label="")
+         output2 = gr.Image(label="")
+         magic1 = gr.Textbox(label="Generated Prompt", lines=2)
+         magic2 = gr.Textbox(label="Generated Prompt", lines=2)
+            
+         run.click(send_it, inputs=[magic1, model_name1], outputs=[output1])
+         run.click(send_it, inputs=[magic2, model_name1], outputs=[output2])   
+        
+    model_name1.change(set_model, inputs=model_name1, outputs=[output1, output2,])
         
     with gr.Tab("Flip Image"):
         with gr.Row():