Fix Zombie type
Browse files
app.py
CHANGED
|
@@ -104,7 +104,7 @@ def inference(img, model):
|
|
| 104 |
my_sample = generatorhulk(my_w, input_is_latent=True)
|
| 105 |
elif model == 'Zombie':
|
| 106 |
with torch.no_grad():
|
| 107 |
-
my_sample =
|
| 108 |
elif model == 'JoJo':
|
| 109 |
with torch.no_grad():
|
| 110 |
my_sample = generatorjojo(my_w, input_is_latent=True)
|
|
|
|
| 104 |
my_sample = generatorhulk(my_w, input_is_latent=True)
|
| 105 |
elif model == 'Zombie':
|
| 106 |
with torch.no_grad():
|
| 107 |
+
my_sample = generatorzombie(my_w, input_is_latent=True)
|
| 108 |
elif model == 'JoJo':
|
| 109 |
with torch.no_grad():
|
| 110 |
my_sample = generatorjojo(my_w, input_is_latent=True)
|