moflo
commited on
Commit
·
e6dd44f
1
Parent(s):
fcd4310
Typo on pretrained download
Browse files
app.py
CHANGED
@@ -563,7 +563,7 @@ print("Loading...")
|
|
563 |
url = "https://github.com/yarri-oss/stylegan_performance/releases/download/v0.1/stylegan_64x64.ckpt.zip"
|
564 |
|
565 |
weights_path = keras.utils.get_file(
|
566 |
-
"stylegan_64x64.ckpt
|
567 |
url,
|
568 |
extract=True,
|
569 |
cache_dir=os.path.abspath("."),
|
@@ -586,7 +586,7 @@ class InferenceWrapper:
|
|
586 |
def __init__(self, model):
|
587 |
self.model = model
|
588 |
self.style_gan = StyleGAN(start_res=START_RES, target_res=TARGET_RES)
|
589 |
-
self.style_gan.grow_model(
|
590 |
self.style_gan.load_weights(os.path.join("pretrained/stylegan_64x64.ckpt"))
|
591 |
self.seed = -1
|
592 |
|
|
|
563 |
url = "https://github.com/yarri-oss/stylegan_performance/releases/download/v0.1/stylegan_64x64.ckpt.zip"
|
564 |
|
565 |
weights_path = keras.utils.get_file(
|
566 |
+
"stylegan_64x64.ckpt",
|
567 |
url,
|
568 |
extract=True,
|
569 |
cache_dir=os.path.abspath("."),
|
|
|
586 |
def __init__(self, model):
|
587 |
self.model = model
|
588 |
self.style_gan = StyleGAN(start_res=START_RES, target_res=TARGET_RES)
|
589 |
+
self.style_gan.grow_model(64)
|
590 |
self.style_gan.load_weights(os.path.join("pretrained/stylegan_64x64.ckpt"))
|
591 |
self.seed = -1
|
592 |
|