Spaces:
Running
Running
Update app.py
Browse files
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 =
|
64 |
model.eval()
|
65 |
-
if seed
|
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")
|