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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -60,9 +60,9 @@ def sample_with_temperature(probs, temperature=1.2):
60
  sampled = sampled.view(batch, height, width)
61
  return sampled
62
 
63
- def generate_map(seed: int = None):
64
  model.eval()
65
- if seed is None:
66
  seed = torch.randint(10000, (1,)).item()
67
  torch.manual_seed(seed)
68
  z = torch.randn(1, model.latent_dim).to("cpu")
 
60
  sampled = sampled.view(batch, height, width)
61
  return sampled
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")