abreza commited on
Commit
2d9fe25
1 Parent(s): f21e8dd

Update gradio_app/gradio_3dgen.py

Browse files
Files changed (1) hide show
  1. gradio_app/gradio_3dgen.py +2 -2
gradio_app/gradio_3dgen.py CHANGED
@@ -7,7 +7,7 @@ from gradio_app.utils import clean_up
7
  from gradio_app.custom_models.mvimg_prediction import run_mvprediction
8
  from gradio_app.custom_models.normal_prediction import predict_normals
9
  from scripts.refine_lr_to_sr import run_sr_fast
10
- from scripts.utils import save_glb_and_video
11
  # from scripts.multiview_inference import geo_reconstruct
12
  from scripts.multiview_inference import geo_reconstruct_part1, geo_reconstruct_part2, geo_reconstruct_part3
13
 
@@ -38,7 +38,7 @@ def generate3dv2(preview_img, input_processing, seed, render_video=True, do_refi
38
  vertices[..., [0, 2]] = - vertices[..., [0, 2]]
39
  new_meshes = Meshes(verts=[vertices], faces=new_meshes.faces_list(), textures=new_meshes.textures)
40
 
41
- ret_mesh, video = save_glb_and_video("/tmp/gradio/generated", new_meshes, with_timestamp=True, dist=3.5, fov_in_degrees=2 / 1.35, cam_type="ortho", export_video=render_video)
42
  return ret_mesh, video
43
 
44
  #######################################
 
7
  from gradio_app.custom_models.mvimg_prediction import run_mvprediction
8
  from gradio_app.custom_models.normal_prediction import predict_normals
9
  from scripts.refine_lr_to_sr import run_sr_fast
10
+ from scripts.utils import save_obj_and_video
11
  # from scripts.multiview_inference import geo_reconstruct
12
  from scripts.multiview_inference import geo_reconstruct_part1, geo_reconstruct_part2, geo_reconstruct_part3
13
 
 
38
  vertices[..., [0, 2]] = - vertices[..., [0, 2]]
39
  new_meshes = Meshes(verts=[vertices], faces=new_meshes.faces_list(), textures=new_meshes.textures)
40
 
41
+ ret_mesh, video = save_obj_and_video("/tmp/gradio/generated", new_meshes, with_timestamp=True, dist=3.5, fov_in_degrees=2 / 1.35, cam_type="ortho", export_video=render_video)
42
  return ret_mesh, video
43
 
44
  #######################################