Update app.py

app.py

@@ -62,22 +62,23 @@ def run_schnell(prompt):
     return image
 
 def run_parallel_models(prompt):
-    print(prompt)
     with ProcessPoolExecutor(max_workers=3) as executor:
         future_dev_hyper = executor.submit(run_dev_hyper, prompt)
         future_dev_turbo = executor.submit(run_dev_turbo, prompt)
         future_schnell = executor.submit(run_schnell, prompt)
         
         res_dev_hyper = future_dev_hyper.result()
+        yield res_dev_hyper, gr.update(), gr.update()
         res_dev_turbo = future_dev_turbo.result()
+        yield gr.update(), res_dev_turbo, gr.update()
         res_schnell = future_schnell.result()
-    return res_dev_hyper, res_dev_turbo, res_schnell
+        yield gr.update(), gr.update(), res_dev_turbo
 
 run_parallel_models.zerogpu = True
 
 with gr.Blocks() as demo:
     gr.Markdown("# Low Step Flux Comparison")
-    gr.Markdown("Compare the quality (not the speed) of FLUX Schnell (4 steps), FLUX.1[dev] HyperFLUX (8 steps), FLUX.1[dev]-Turbo-Alpha (8 steps)")
+    gr.Markdown("Compare the quality (not the speed) of FLUX Schnell (4 steps), FLUX.1[dev] HyperFLUX (8 steps), FLUX.1[dev]-Turbo-Alpha (8 steps). It runs a bit slow as it's inferencing the three models.")
     with gr.Row():
         with gr.Column(scale=2):
             prompt = gr.Textbox(label="Prompt")
@@ -89,14 +90,21 @@ with gr.Blocks() as demo:
         turbo = gr.Image(label="FLUX.1[dev]-Turbo-Alpha (8 steps)")
     
     gr.Examples(
-        examples=[["the spirit of a Tamagotchi wandering in the city of Vienna"], ["a photo of a lavender cat"], ["a tiny astronaut hatching from an egg on the moon"], ["A delicious ceviche cheesecake slice"], ["an insect robot preparing a delicious meal"], ["A Charmander fine dining with a view to la Sagrada Família"]],
+        examples=[
+            ["the spirit of a Tamagotchi wandering in the city of Vienna"],
+            ["a photo of a lavender cat"],
+            ["a tiny astronaut hatching from an egg on the moon"],
+            ["a delicious ceviche cheesecake slice"],
+            ["an insect robot preparing a delicious meal"],
+            ["a Charmander fine dining with a view to la Sagrada Família"]],
         fn=run_parallel_models,
         inputs=[prompt],
         outputs=[schnell, hyper, turbo],
         cache_examples="lazy"
     )
     
-    submit.click(
+    gr.on(
+        triggers=[submit.click, prompt.submit],
         fn=run_parallel_models,
         inputs=[prompt],
         outputs=[schnell, hyper, turbo]