FrozenBurning commited on
Commit
e1329d4
·
1 Parent(s): a4912eb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -137,7 +137,7 @@ def process(input_cond, input_num_steps, input_seed=42, input_cfg=6.0):
137
  recon_feat_param[:, 1:, ...] = (recon_feat_param[:, 1:, ...] + 1) / 2.
138
  recon_feat_param = recon_feat_param.reshape(inf_bs, config.model.num_prims, -1)
139
  recon_param = torch.concat([recon_srt_param, recon_feat_param], dim=-1)
140
- visualize_video_primvolume(config.output_dir, batch, recon_param, 30, rm, device)
141
  prim_params = {'srt_param': recon_srt_param[0].detach().cpu(), 'feat_param': recon_feat_param[0].detach().cpu()}
142
  torch.save({'model_state_dict': prim_params}, "{}/denoised.pt".format(config.output_dir))
143
 
@@ -200,7 +200,7 @@ with block:
200
  # gen button
201
  button_gen = gr.Button(value="Generate", interactive=False)
202
  with gr.Row():
203
- input_mc_resolution = gr.Radio(choices=[32, 64, 128, 256], label="MC Resolution", value=128, info="Cube resolution for mesh extraction")
204
  input_remesh = gr.Radio(choices=["No", "Yes"], label="Remesh", value="No", info="Remesh or not?")
205
  export_glb_btn = gr.Button(value="Export GLB", interactive=False)
206
 
 
137
  recon_feat_param[:, 1:, ...] = (recon_feat_param[:, 1:, ...] + 1) / 2.
138
  recon_feat_param = recon_feat_param.reshape(inf_bs, config.model.num_prims, -1)
139
  recon_param = torch.concat([recon_srt_param, recon_feat_param], dim=-1)
140
+ visualize_video_primvolume(config.output_dir, batch, recon_param, 15, rm, device)
141
  prim_params = {'srt_param': recon_srt_param[0].detach().cpu(), 'feat_param': recon_feat_param[0].detach().cpu()}
142
  torch.save({'model_state_dict': prim_params}, "{}/denoised.pt".format(config.output_dir))
143
 
 
200
  # gen button
201
  button_gen = gr.Button(value="Generate", interactive=False)
202
  with gr.Row():
203
+ input_mc_resolution = gr.Radio(choices=[64, 128, 256], label="MC Resolution", value=128, info="Cube resolution for mesh extraction")
204
  input_remesh = gr.Radio(choices=["No", "Yes"], label="Remesh", value="No", info="Remesh or not?")
205
  export_glb_btn = gr.Button(value="Export GLB", interactive=False)
206