Update app.py

app.py

@@ -57,11 +57,19 @@ pipe_hyper.to("cuda")
 del unet
 
 @spaces.GPU
-def run_comparison(prompt, progress=gr.Progress(track_tqdm=True)):
+def run_comparison_turbo(prompt, progress=gr.Progress(track_tqdm=True)):
     image_turbo=pipe_turbo(prompt=prompt, num_inference_steps=1, guidance_scale=0).images[0]
+    return image_turbo
+
+@spaces.GPU
+def run_comparison_lightning(prompt, progress=gr.Progress(track_tqdm=True)):
     image_lightning=pipe_lightning(prompt=prompt, num_inference_steps=1, guidance_scale=0).images[0]
+    return image_lightning
+
+@spaces.GPU
+def run_comparison_hyper(prompt, progress=gr.Progress(track_tqdm=True)):
     image_hyper=pipe_hyper(prompt=prompt, num_inference_steps=1, guidance_scale=0, timesteps=[800]).images[0]
-    return image_turbo, image_lightning, image_hyper
+    return image_hyper
 
 examples = ["A dignified beaver wearing glasses, a vest, and colorful neck tie.",
 "The spirit of a tamagotchi wandering in the city of Barcelona",
@@ -90,16 +98,23 @@ with gr.Blocks() as demo:
     image_outputs = [image_turbo, image_lightning, image_hyper]
     gr.on(
         triggers=[prompt.submit, run.click],
-        fn=run_comparison,
+        fn=run_comparison_turbo,
+        inputs=prompt,
+        outputs=image_turbo
+    ).then(
+        fn=run_comparison_lightning,
+        inputs=prompt,
+        outputs=image_lightning
+    ).then(
+        fn=run_comparison_hyper,
         inputs=prompt,
-        outputs=image_outputs
+        outputs=image_hyper
     )
     gr.Examples(
         examples=examples,
         fn=run_comparison,
         inputs=prompt,
         outputs=image_outputs,
-        cache_examples=False,
-        run_on_click=True
+        cache_examples=False
     )
 demo.launch()