Streaming

app.py

@@ -75,7 +75,7 @@ def generate_image(prompt: str, progress=gr.Progress(track_tqdm=True)):
         image = pipeline(prompt, num_inference_steps=4).images[0]
         elapsed = -(t0 - (t0 := datetime.now()))
         images += [(image, f'{elapsed.total_seconds():.2f}s')]
-    return images
+        yield images
 
 
 gr.Interface(generate_image, gr.Text(), gr.Gallery(rows=3, columns=3, height='60vh')).launch()