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 |
|