Update app.py

app.py

@@ -22,8 +22,9 @@ colorizer, colorLabeler = setup_model(checkpt_path, device=device)
 def click_colorize(rgb_img, hint_img, n_anchors, is_high_res, is_editable):
     if hint_img is None:
         hint_img = rgb_img
-    output = colorize_grayscale(colorizer, colorLabeler, rgb_img, hint_img, n_anchors, is_high_res, is_editable, device)
-    return output
+    output = colorize_grayscale(colorizer, colorLabeler, rgb_img, hint_img, n_anchors, True, is_editable, device)
+    output1 = colorize_grayscale(colorizer, colorLabeler, rgb_img, hint_img, n_anchors, False, is_editable, device)
+    return output, output1
 
 def click_predanchors(rgb_img, n_anchors, is_high_res, is_editable):
     output = predict_anchors(colorizer, colorLabeler, rgb_img, n_anchors, is_high_res, is_editable, device)
@@ -55,7 +56,7 @@ with demo:
                 Button_show_anchor = gr.Button(value="Predict anchors", visible=False)
             Button_run = gr.Button(value="Colorize")
         with gr.Column():
-            Image_output = gr.Image(type="numpy", label="Output").style(height=480)
+            Image_output = [gr.Image(type="numpy", label="Output").style(height=480), gr.Image(type="numpy", label="Output").style(height=480)]
 
     Ckeckbox_editable.change(fn=switch_states, inputs=Ckeckbox_editable, outputs=[Image_anchor, Button_show_anchor])
     Button_show_anchor.click(fn=click_predanchors, inputs=[Image_input, Num_anchor, Radio_resolution, Ckeckbox_editable], outputs=Image_anchor)
@@ -78,7 +79,7 @@ with demo:
                     ['03.jpg', 8, "Low (256x256)"],
                     ['04.jpg', 8, "Low (256x256)"],
                     ], 
-                    inputs=[Image_input,Num_anchor,Radio_resolution], outputs=[Image_output], label="Examples")
+                    inputs=[Image_input,Num_anchor,Radio_resolution], outputs=[Image_output], label="Examples", cache_examples=False)
     gr.HTML(value="""
                 <p style="text-align:center; color:orange"><a href='https://menghanxia.github.io/projects/disco.html' target='_blank'>DISCO Project Page</a> | <a href='https://github.com/MenghanXia/DisentangledColorization' target='_blank'>Github Repo</a></p>
                     """)