Update app.py

app.py

@@ -14,7 +14,7 @@ from inference import Predictor
 
 def parse_args():
     parser = argparse.ArgumentParser()
-    parser.add_argument('--weight', type=str, default='cartoonize/GeneratorV2_train_photo_Hayao_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()
@@ -38,11 +38,6 @@ def clear_output(input_widget):
     input_widget = np.array([])
 
 with gr.Blocks() as demo:
-    os.system("wget https://huggingface.co/YANGYYYY/cartoonize/tree/main/GeneratorV2_train_photo_Hayao_init.pt")
-    if os.path.exists("GeneratorV2_train_photo_Hayao_init.pt"):
-        print("下载成功！")
-    else:
-        print("下载失败！")
     gr.Markdown("Transfer image or video files using this demo.")
     with gr.Tabs():
         with gr.TabItem("Transfer Image"):