app

app.py

@@ -174,20 +174,22 @@ def text_to_image(prompt, seed=None, strength=1.0,lora_scale=1.0, num_inference_
                                       **kwargs)
     return result[-1]
 
+@spaces.GPU(duration=120)
 def image2mesh_preprocess_(input_image_, seed, use_mv_rgb=True):
     global preprocessed_input_image
 
     seed = int(seed) if seed is not None else None
 
     # TODO: delete this later
-    k3d_wrapper.del_llm_model()
+    # k3d_wrapper.del_llm_model()
     
     input_image_save_path, reference_save_path, caption = image2mesh_preprocess(k3d_wrapper, input_image_, seed, use_mv_rgb)
 
     preprocessed_input_image = Image.open(input_image_save_path)
     return reference_save_path, caption
 
-@spaces.GPU
+
+@spaces.GPU(duration=120)
 def image2mesh_main_(reference_3d_bundle_image, caption, seed, strength1=0.5, strength2=0.95, enable_redux=True, use_controlnet=True, if_video=True):
     subprocess.run(['nvidia-smi'])  
     global mesh_cache 
@@ -195,7 +197,7 @@ def image2mesh_main_(reference_3d_bundle_image, caption, seed, strength1=0.5, st
 
 
     # TODO: delete this later
-    k3d_wrapper.del_llm_model()
+    # k3d_wrapper.del_llm_model()
 
     input_image = preprocessed_input_image
 
@@ -205,8 +207,6 @@ def image2mesh_main_(reference_3d_bundle_image, caption, seed, strength1=0.5, st
     mesh_cache = recon_mesh_path
 
 
-    # gen_save_ = Image.open(gen_save_path)
-
     if if_video:
         video_path = recon_mesh_path.replace('.obj','.mp4').replace('.glb','.mp4')
         render_video_from_obj(recon_mesh_path, video_path)
@@ -214,7 +214,6 @@ def image2mesh_main_(reference_3d_bundle_image, caption, seed, strength1=0.5, st
         return gen_save_path, video_path
     else:
         return gen_save_path, recon_mesh_path
-    # return gen_save_path, recon_mesh_path
 
 @spaces.GPU(duration=120)
 def bundle_image_to_mesh(
@@ -422,7 +421,7 @@ with gr.Blocks(css="""
 
                     btn_gen_mesh = gr.Button("Generate Mesh")
                     output_video1 = gr.Video(label="Generated Video", interactive=False, loop=True, autoplay=True)
-                    btn_download1 = gr.Button("Download Mesh")
+                    # btn_download1 = gr.Button("Download Mesh")
 
 
                     download_1 = gr.DownloadButton(label="Download mesh", interactive=False)
@@ -479,16 +478,11 @@ with gr.Blocks(css="""
         lambda: gr.Button(interactive=True),
         outputs=[download_1],
     )
-    # btn_gen_mesh.click(fn=bundle_image_to_mesh, inputs=[output_image1, lrm_radius, isomer_radius, reconstruction_stage1_steps, reconstruction_stage2_steps], outputs=output_video1)
-
     with gr.Row():
         pass
     with gr.Row():
         gr.Markdown(_CITE_)
 
-# demo.queue(default_concurrency_limit=1)
-# demo.launch(server_name="0.0.0.0", server_port=9239)
-# subprocess.run("rm -rf /data-nvme/zerogpu-offload/*", env={}, shell=True)
 demo.launch()