Update app.py

app.py

@@ -19,6 +19,7 @@ device = "cpu"
 checkpt_path = "checkpoints/disco-beta.pth.rar"
 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
@@ -26,10 +27,12 @@ def click_colorize(rgb_img, hint_img, n_anchors, is_high_res, is_editable):
     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)
     return output
 
+
 ## step 2: configure interface
 def switch_states(is_checked):
     if is_checked:
@@ -37,6 +40,7 @@ def switch_states(is_checked):
     else:
         return gr.Image.update(visible=False), gr.Button.update(visible=False)
 
+
 demo = gr.Blocks(title="DISCO", delete_cache=(1800, 3600),)
 with demo:
     gr.Markdown(value="""
@@ -84,6 +88,7 @@ with demo:
                 <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>
                     """)
 
+
 if RUN_MODE == "local":
     demo.launch(server_name='9.134.253.83',server_port=7788)
 else: