Spaces:
Running
Running
Update app.py
Browse files
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
|
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")
|