Update app.py

app.py

@@ -38,7 +38,14 @@ vqvae_model = vqvae.HumanVQVAE(args,
                        args.width,
                        args.depth,
                        args.dilation_growth_rate).to(device)
-transformer_model = trans.Text2Motion_Transformer().to(device)
+transformer_model = trans.Text2Motion_Transformer(num_vq=args.nb_code,
+                                embed_dim=1024,
+                                clip_dim=args.clip_dim,
+                                block_size=args.block_size,
+                                num_layers=9,
+                                n_head=16,
+                                drop_out_rate=args.drop_out_rate,
+                                fc_rate=args.ff_rate).to(device)
 vqvae_model.load_state_dict(torch.load("output/VQVAE_imp_resnet_100k_hml3d/net_last.pth", map_location=device))
 transformer_model.load_state_dict(torch.load("output/net_best_fid.pth", map_location=device))
 vqvae_model.eval()