added some line to show progress

app.py

@@ -1,9 +1,35 @@
 from diffusers import DiffusionPipeline
 import gradio as gr
+import sys 
 
 generator = DiffusionPipeline.from_pretrained("kaveh/wsi_generator")
 
-def generate(n_samples=1):
+class Logger:
+    def __init__(self, filename):
+        self.terminal = sys.stdout
+        self.log = open(filename, "w")
+
+    def write(self, message):
+        self.terminal.write(message)
+        self.log.write(message)
+        
+    def flush(self):
+        self.terminal.flush()
+        self.log.flush()
+        
+    def isatty(self):
+        return False  
+        
+sys.stdout = Logger("output.log")
+
+def read_logs():
+    sys.stdout.flush()
+    with open("output.log", "r") as f:
+        return f.read()
+
+  
+
+def generate(n_samples=1, progress=gr.Progress()):
     images = []
     for i in range(n_samples):
         image = generator().images[0]
@@ -20,5 +46,7 @@ with gr.Blocks() as demo:
             label="Generated images", show_label=False, elem_id="gallery").style(columns=[2], rows=[2], object_fit="contain", height="auto", preview=True)
 
     btn.click(generate, n_s, gallery)
-
+    logs = gr.Textbox()
+    demo.load(read_logs, None, logs, every=1)
+    
 demo.launch()