Spaces:
Running
Running
Update app.py
Browse files
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/
|
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()
|