Update app.py

app.py

@@ -57,19 +57,13 @@ pipe_hyper.to("cuda")
 del unet
 
 @spaces.GPU
-def run_comparison_turbo(prompt, progress=gr.Progress(track_tqdm=True)):
+def run_comparison(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)):
+    yield image_turbo, None, None
     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)):
+    yield image_turbo, image_lightning, None
     image_hyper=pipe_hyper(prompt=prompt, num_inference_steps=1, guidance_scale=0, timesteps=[800]).images[0]
-    return image_hyper
+    return image_turbo, image_lightning, 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",
@@ -98,22 +92,16 @@ with gr.Blocks() as demo:
     image_outputs = [image_turbo, image_lightning, image_hyper]
     gr.on(
         triggers=[prompt.submit, run.click],
-        fn=run_comparison_turbo,
-        inputs=prompt,
-        outputs=image_turbo
-    ).then(
-        fn=run_comparison_lightning,
-        inputs=prompt,
-        outputs=image_lightning
-    ).then(
-        fn=run_comparison_hyper,
+        fn=run_comparison,
         inputs=prompt,
-        outputs=image_hyper
+        outputs=image_outputs
     )
     gr.Examples(
         examples=examples,
+        fn=run_comparison,
         inputs=prompt,
         outputs=image_outputs,
-        cache_examples=False
+        cache_examples=False,
+        run_on_click=True
     )
 demo.launch()