Ahsen Khaliq
commited on
Commit
·
747cb4e
1
Parent(s):
0562394
Update app.py
Browse files
app.py
CHANGED
|
@@ -55,7 +55,7 @@ mean_latentchurch = load_model(generator, 'church.pt')
|
|
| 55 |
def inferece(num, seed, model_type):
|
| 56 |
if model_type == 'landscape':
|
| 57 |
mean_latent = mean_latentland
|
| 58 |
-
elif model_type
|
| 59 |
mean_latent = mean_latentface
|
| 60 |
else:
|
| 61 |
mean_latent = mean_latentchurch
|
|
|
|
| 55 |
def inferece(num, seed, model_type):
|
| 56 |
if model_type == 'landscape':
|
| 57 |
mean_latent = mean_latentland
|
| 58 |
+
elif model_type == 'face':
|
| 59 |
mean_latent = mean_latentface
|
| 60 |
else:
|
| 61 |
mean_latent = mean_latentchurch
|