Spaces:
Runtime error
Runtime error
SerdarHelli
commited on
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")
|