sztanki commited on
Commit
59506ce
·
1 Parent(s): e6ef7fa

Fix Zombie type

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = generatorjojo(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)
 
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)