jamino30 commited on
Commit
e4d04bf
·
verified ·
1 Parent(s): ead29c9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -62,7 +62,7 @@ def sample_with_temperature(probs, temperature=1.2):
62
 
63
  def generate_map(seed: int = 0):
64
  model.eval()
65
- if seed != 0:
66
  seed = torch.randint(10000, (1,)).item()
67
  torch.manual_seed(seed)
68
  z = torch.randn(1, model.latent_dim).to("cpu")
 
62
 
63
  def generate_map(seed: int = 0):
64
  model.eval()
65
+ if seed == 0:
66
  seed = torch.randint(10000, (1,)).item()
67
  torch.manual_seed(seed)
68
  z = torch.randn(1, model.latent_dim).to("cpu")