Update app.py

app.py

@@ -8,7 +8,7 @@ from inference import Predictor
 
 def parse_args():
     parser = argparse.ArgumentParser()
-    parser.add_argument('--weight', type=str, default='GeneratorV2_train_photo_Paprika_init.pt')
+    parser.add_argument('--weight', type=str, default='GeneratorV2_train_photo_Hayao_init.pt')
     parser.add_argument('--device', type=str, default='cpu', help='Device, cuda or cpu')
 
     return parser.parse_args()
@@ -16,6 +16,7 @@ def parse_args():
 def transfer(image, transfer_style):
     if transfer_style == "Hayao":
          #output = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)# 转换为灰度图像
+        os.system("wget https://huggingface.co/YANGYYYY/cartoonize/resolve/main/GeneratorV2_train_photo_Hayao_init.pt")
         predictor = Predictor(args.weight, args.device)
         anime_img = predictor.transform_image(args.src, save_path)
         return anime_img