# This is a combination of 2 commits.

# This is the 1st commit message:

init

# The commit message #2 will be skipped:

# # This is a combination of 2 commits.
# # This is the 1st commit message:
#
# init
#
# # The commit message #2 will be skipped:
#
# # # This is a combination of 4 commits.
# # # This is the 1st commit message:
# #
# # init
# #
# # # The commit message #2 will be skipped:
# #
# # # init
# #
# # # The commit message #3 will be skipped:
# #
# # # init
# #
# # # The commit message #4 will be skipped:
# #
# # # init

.gitattributes

@@ -33,4 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
-*.png* filter=lfs diff=lfs merge=lfs -text
+*.png* filter=lfs diff=lfs merge=lfs -text
+*.whl filter=lfs diff=lfs merge=lfs -text

app.py

@@ -6,8 +6,12 @@ from gradio_model4dgs import Model4DGS
 import numpy
 import hashlib
 
-# os.system('git clone --recursive https://github.com/ashawkey/diff-gaussian-rasterization && pip install -e ./diff-gaussian-rasterization')
-os.system('pip install xformers==0.0.23 --no-deps --index-url https://download.pytorch.org/whl/cu118')
+import spaces
+
+@spaces.GPU
+def install():
+    subprocess.run(shlex.split("pip install wheel/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl"))
+    os.system('pip install xformers==0.0.23 --no-deps --index-url https://download.pytorch.org/whl/cu118')
 
 from huggingface_hub import hf_hub_download
 ckpt_path = hf_hub_download(repo_id="ashawkey/LGM", filename="model_fp16_fixrot.safetensors")
@@ -34,7 +38,7 @@ def check_video_input(image_block: Image.Image):
     if not os.path.exists(os.path.join('tmp_data', f'{img_hash}_rgba_generated.mp4')):
         raise gr.Error("Please generate a video first")
 
-
+@spaces.GPU
 def optimize_stage_1(image_block: Image.Image, preprocess_chk: bool, seed_slider: int):
     if not os.path.exists('tmp_data'):
         os.makedirs('tmp_data')
@@ -55,7 +59,7 @@ def optimize_stage_1(image_block: Image.Image, preprocess_chk: bool, seed_slider
     # return [os.path.join('logs', 'tmp_rgba_model.ply')]
     return os.path.join('tmp_data', f'{img_hash}_rgba_generated.mp4')
 
-
+@spaces.GPU
 def optimize_stage_2(image_block: Image.Image, seed_slider: int):
     img_hash = hashlib.sha256(image_block.tobytes()).hexdigest()
     subprocess.run(f'python lgm/infer.py big --resume {ckpt_path} --test_path tmp_data/{img_hash}_rgba.png', shell=True)
@@ -86,6 +90,8 @@ if __name__ == "__main__":
     example_fns.sort()
     examples_full = [os.path.join(example_folder, x) for x in example_fns if x.endswith('.png')]
 
+    install()
+
     # Compose demo layout & data flow
     with gr.Blocks(title=_TITLE, theme=gr.themes.Soft(), js=js_func) as demo:
         with gr.Row():
@@ -132,4 +138,4 @@ if __name__ == "__main__":
 
     # demo.queue().launch(share=True)
     demo.queue(max_size=10)  # <-- Sets up a queue with default parameters
-    demo.launch(share=True)
+    demo.launch()

requirements.txt

@@ -30,5 +30,7 @@ rembg[gpu,cli]
 # gradio demo
 gradio
 gradio-model4dgs
+tyro
+roma
 
 -e git+https://github.com/ashawkey/kiuikit.git@main#egg=kiui