Update app.py

app.py

@@ -297,9 +297,9 @@ with gr.Blocks() as demo:
             run = gr.Button('Generate')
 
         with gr.Column():
-            result0 = gr.Gallery([(None,"Initial Word"),(None,"Optimization Process"),(None,"Final Result")], label="Output",columns=1)#gr.HTML("""<style>{width: 333px; height: 333px;}</style>""",label="Initial Word") #gr.Image(type="filepath", label="Initial Word",height=333) 
-            # result1 = gr.Image(type="filepath", label="Optimization Process",height=100) 
-            # result2 = gr.HTML(label="Final Result") #gr.Image(type="filepath", label="Final Result",visible=False,height=333) 
+            result0 = gr.Gallery([(None,"Initial Word")], label="Initial Word")#gr.HTML("""<style>{width: 333px; height: 333px;}</style>""",label="Initial Word") #gr.Image(type="filepath", label="Initial Word",height=333) 
+            result1 = gr.Gallery([(None,"Optimization Process")], label="Optimization Process")#gr.Image(type="filepath", label="Optimization Process",height=100) 
+            result2 = gr.Gallery([(None,"Final Result")], label="Final Result")#gr.HTML(label="Final Result") #gr.Image(type="filepath", label="Final Result",visible=False,height=333) 
             
         
     with gr.Row():
@@ -343,9 +343,9 @@ with gr.Blocks() as demo:
                     num_steps
                 ],
                 outputs=[
-                    result0#,
-                    # result1,
-                    # result2
+                    result0,
+                    result1,
+                    result2
                 ],
                 fn=run_main_ex,
                 cache_examples=True)
@@ -361,9 +361,9 @@ with gr.Blocks() as demo:
     ]
 
     outputs = [
-        result0#,
-        # result1,
-        # result2
+        result0,
+        result1,
+        result2
     ]
     
     run.click(fn=run_main_app, inputs=inputs, outputs=outputs, concurrency_limit=2)