Update app.py

app.py

@@ -9,7 +9,9 @@ if RUN_MODE != "local":
     os.system("wget https://huggingface.co/menghanxia/disco/resolve/main/disco-beta.pth.rar")
     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")
 
 ## step 1: set up model
@@ -20,9 +22,8 @@ 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, True, is_editable, device)
-    output1 = colorize_grayscale(colorizer, colorLabeler, rgb_img, hint_img, n_anchors, False, is_editable, device)
-    return output, output1
+    output = colorize_grayscale(colorizer, colorLabeler, rgb_img, hint_img, n_anchors, is_high_res, is_editable, device)
+    return output
 
 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)