Update gradio_app.py

gradio_app.py

@@ -42,10 +42,10 @@ _CITE_ = r"""
 📝 **Citation**
 If you find our work useful for your research or applications, please cite using this bibtex:
 ```bibtex
-@article{craftsman,
+@article{li2024craftsman,
 author    = {Weiyu Li and Jiarui Liu and Rui Chen and Yixun Liang and Xuelin Chen and Ping Tan and Xiaoxiao Long},
 title     = {CraftsMan: High-fidelity Mesh Generation with 3D Native Generation and Interactive Geometry Refiner},
-journal   = {arxiv:xxx},
+journal   = {arXiv preprint arXiv:2405.14979},
 year      = {2024},
 }
 ```
@@ -191,10 +191,12 @@ if __name__=="__main__":
     )
 
     # for 3D latent set diffusion
-    ckpt_path = "./ckpts/image-to-shape-diffusion/clip-mvrgb-modln-l256-e64-ne8-nd16-nl6/model.ckpt"
-    config_path = "./ckpts/image-to-shape-diffusion/clip-mvrgb-modln-l256-e64-ne8-nd16-nl6/config.yaml"
+    # ckpt_path = "./ckpts/image-to-shape-diffusion/clip-mvrgb-modln-l256-e64-ne8-nd16-nl6/model.ckpt"
+    # config_path = "./ckpts/image-to-shape-diffusion/clip-mvrgb-modln-l256-e64-ne8-nd16-nl6/config.yaml"
     # ckpt_path = hf_hub_download(repo_id="wyysf/CraftsMan", filename="image-to-shape-diffusion/clip-mvrgb-modln-l256-e64-ne8-nd16-nl6/model.ckpt", repo_type="model")
     # config_path = hf_hub_download(repo_id="wyysf/CraftsMan", filename="image-to-shape-diffusion/clip-mvrgb-modln-l256-e64-ne8-nd16-nl6/config.yaml", repo_type="model")
+    ckpt_path = hf_hub_download(repo_id="wyysf/CraftsMan", filename="image-to-shape-diffusion/clip-mvrgb-modln-l256-e64-ne8-nd16-nl6-aligned-vae/model-300k.ckpt", repo_type="model")
+    config_path = hf_hub_download(repo_id="wyysf/CraftsMan", filename="image-to-shape-diffusion/clip-mvrgb-modln-l256-e64-ne8-nd16-nl6-aligned-vae/config.yaml", repo_type="model")
     scheluder_dict = OrderedDict({ 
         "DDIMScheduler": 'diffusers.schedulers.DDIMScheduler',
         # "DPMSolverMultistepScheduler": 'diffusers.schedulers.DPMSolverMultistepScheduler', # not support yet
@@ -289,12 +291,12 @@ if __name__=="__main__":
                         # backgroud_color = gr.ColorPicker(label="Background Color", value="#7F7F7F", interactive=True)
                         
                     with gr.Row():
-                        mvimg_guidance_scale = gr.Number(value=4.0, minimum=3, maximum=10, label="2D Guidance Scale")
+                        mvimg_guidance_scale = gr.Number(value=3.0, minimum=1, maximum=10, label="2D Guidance Scale")
                         mvimg_steps = gr.Number(value=30, minimum=20, maximum=100, label="2D Sample Steps")
             
                 with gr.Accordion('Advanced options (3D)', open=False):
                     with gr.Row():
-                        guidance_scale = gr.Number(label="3D Guidance Scale", value=7.5, minimum=3.0, maximum=10.0)
+                        guidance_scale = gr.Number(label="3D Guidance Scale", value=5.0, minimum=1.0, maximum=10.0)
                         steps = gr.Number(value=50, minimum=20, maximum=100, label="3D Sample Steps")
                         
                     with gr.Row():