Commit
·
88949c2
1
Parent(s):
933cc55
Update app.py
Browse files
app.py
CHANGED
|
@@ -63,7 +63,7 @@ def predict(Seed,ckpt,truncate_std):
|
|
| 63 |
gen_pcs = []
|
| 64 |
with torch.no_grad():
|
| 65 |
z = torch.randn([1, ckpt['args'].latent_dim]).to(device)
|
| 66 |
-
x = model.sample(z, 2048, flexibility=ckpt['args'].flexibility,truncate_std=truncate_std)
|
| 67 |
gen_pcs.append(x.detach().cpu())
|
| 68 |
gen_pcs = torch.cat(gen_pcs, dim=0)[:1]
|
| 69 |
gen_pcs = normalize_point_clouds(gen_pcs, mode="shape_bbox")
|
|
|
|
| 63 |
gen_pcs = []
|
| 64 |
with torch.no_grad():
|
| 65 |
z = torch.randn([1, ckpt['args'].latent_dim]).to(device)
|
| 66 |
+
x = model.sample(z, 2048, flexibility=ckpt['args'].flexibility,truncate_std=truncate_std)
|
| 67 |
gen_pcs.append(x.detach().cpu())
|
| 68 |
gen_pcs = torch.cat(gen_pcs, dim=0)[:1]
|
| 69 |
gen_pcs = normalize_point_clouds(gen_pcs, mode="shape_bbox")
|