SerdarHelli commited on
Commit
88949c2
1 Parent(s): 933cc55

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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")