YANGYYYY commited on
Commit
385ddd4
·
verified ·
1 Parent(s): 7767bc5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ from inference import Predictor
8
 
9
  def parse_args():
10
  parser = argparse.ArgumentParser()
11
- parser.add_argument('--weight', type=str, default='cartoonize/resolve/main/GeneratorV2_train_photo_Hayao_init.pt')
12
  parser.add_argument('--device', type=str, default='cpu', help='Device, cuda or cpu')
13
 
14
  return parser.parse_args()
 
8
 
9
  def parse_args():
10
  parser = argparse.ArgumentParser()
11
+ parser.add_argument('--weight', type=str, default='cartoonize/GeneratorV2_train_photo_Hayao_init.pt')
12
  parser.add_argument('--device', type=str, default='cpu', help='Device, cuda or cpu')
13
 
14
  return parser.parse_args()