Update app.py

app.py

@@ -178,7 +178,7 @@ with gr.Blocks() as demo:
 
                 with gr.Column():
                     gen_output = gr.Textbox(label="Generation Results", lines=5, placeholder="")
-                    gen_output_image = gr.Textbox(label="Visualization", lines=3, placeholder="")
+                    gen_output_image = gr.Image(type='pil', label="Visualization")
        
             gr.Examples(
                 examples=[["[C][=C][C][=C][C][=C][Ring1][=Branch1]"], 
@@ -197,11 +197,12 @@ with gr.Blocks() as demo:
                     opt_button = gr.Button("Optimize")
 
                 with gr.Column():
-                    opt_input_img = gr.Textbox(label="Input Visualization", lines=3, placeholder="")
+                    opt_input_img = gr.Image(type='pil', label="Input Visualization")
                     opt_output = gr.Textbox(label="Optimization Results", lines=3, placeholder="")
                     opt_output_imp = gr.Textbox(label="Optimization Property Improvements", lines=3, placeholder="")
                     opt_output_sim = gr.Textbox(label="Similarity", lines=3, placeholder="")
                     opt_output_img = gr.Textbox(label="Output Visualization", lines=3, placeholder="")
+                    opt_output_img = gr.Image(type='pil', label="Output Visualization")
 
             gr.Examples(
                 examples=[