Update app.py

app.py

@@ -6,13 +6,13 @@ from inference import setup_model, colorize_grayscale, predict_anchors
 ## local |  remote
 RUN_MODE = "remote"
 if RUN_MODE != "local":
-    os.system("wget https://huggingface.co/menghanxia/disco/resolve/main/disco-beta.pth.rar")
+    os.system("wget https://huggingface.co/menghanxia/disco/resolve/main/disco-beta.pth.rar -q -P")
     os.rename("disco-beta.pth.rar", "./checkpoints/disco-beta.pth.rar")
     ## examples
-    os.system("wget https://huggingface.co/menghanxia/disco/resolve/main/01.jpg")
-    os.system("wget https://huggingface.co/menghanxia/disco/resolve/main/02.jpg")
-    os.system("wget https://huggingface.co/menghanxia/disco/resolve/main/03.jpg")
-    os.system("wget https://huggingface.co/menghanxia/disco/resolve/main/04.jpg")
+    os.system("wget https://huggingface.co/menghanxia/disco/resolve/main/01.jpg -q -P")
+    os.system("wget https://huggingface.co/menghanxia/disco/resolve/main/02.jpg -q -P")
+    os.system("wget https://huggingface.co/menghanxia/disco/resolve/main/03.jpg -q -P")
+    os.system("wget https://huggingface.co/menghanxia/disco/resolve/main/04.jpg -q -P")
 
 ## step 1: set up model
 device = "cpu"
@@ -46,7 +46,7 @@ with demo:
         with gr.Column():
             with gr.Row():
                 Image_input = gr.Image(type="numpy", label="Input", interactive=True)
-                Image_anchor = gr.Image(type="numpy", label="Anchor", tool="color-sketch", interactive=True, visible=False)
+                Image_anchor = gr.Image(type="numpy", label="Anchor", interactive=True, visible=False)
             with gr.Row():
                 Num_anchor = gr.Number(type="int", value=8, label="Num. of anchors (3~14)")
                 Radio_resolution = gr.Radio(type="index", choices=["Low (256x256)", "High (512x512)"], \
@@ -56,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), gr.Image(type="numpy", label="Output")]
+            Image_output = [gr.Image(type="numpy", label="Output"), gr.Image(type="numpy", label="Output")]
 
     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)