Update Model Loading

app.py

@@ -7,6 +7,7 @@ import argparse
 import gradio as gr
 import uuid
 import spaces
+from huggingface_hub import hf_hub_download
 #
 
 subprocess.run(shlex.split("pip install wheel/torch_scatter-2.1.2+pt21cu121-cp310-cp310-linux_x86_64.whl"))
@@ -19,13 +20,21 @@ subprocess.run(shlex.split("pip install wheel/pointops-1.0-cp310-cp310-linux_x86
 from src.utils.visualization_utils import render_video_from_file
 from src.model import LSM_MASt3R
 
-model = LSM_MASt3R.from_pretrained("checkpoints/pretrained_model/checkpoint-40.pth")
+# Assuming your model has been uploaded to HuggingFace
+model_repo = "kairunwen/LSM"  # Replace with the actual repository name
+model_filename = "checkpoint-40.pth"  # Model filename
+
+# Download model from HuggingFace
+model_path = hf_hub_download(repo_id=model_repo, filename=model_filename)
+
+# Load model
+model = LSM_MASt3R.from_pretrained(model_path)
 model = model.eval()
 
 
 @spaces.GPU(duration=80)
 def process(inputfiles, input_path=None):
-    # 创建唯一的缓存目录
+    # Create a unique cache directory
     cache_dir = os.path.join('outputs', str(uuid.uuid4()))
     os.makedirs(cache_dir, exist_ok=True)
 
@@ -43,7 +52,7 @@ def process(inputfiles, input_path=None):
     filelist = inputfiles
     if len(filelist) != 2:
         gr.Warning("Please select 2 images")
-        shutil.rmtree(cache_dir)  # 清理缓存目录
+        shutil.rmtree(cache_dir)  # Clean up cache directory
         return None, None, None, None, None, None
     
     ply_path = os.path.join(cache_dir, 'gaussians.ply')
@@ -96,8 +105,8 @@ with block:
                         show_label=False,
                         elem_id="gallery",
                         columns=[2],
-                        height=300,  # 固定高度
-                        object_fit="cover"  # 确保图片填满空间
+                        height=300,  # Fixed height
+                        object_fit="cover"  # Ensure images fill the space
                     )
         
         button_gen = gr.Button("Start Reconstruction", elem_id="button_gen")
@@ -118,7 +127,7 @@ with block:
                     output_model = gr.Model3D(
                         label="3D Dense Model under Gaussian Splats Formats, need more time to visualize",
                         interactive=False,
-                        camera_position=[0.5, 0.5, 1],  # 稍微偏移一点，以便更好地查看模型
+                        camera_position=[0.5, 0.5, 1],  # Slight offset for better model viewing
                         height=600,
                     )
                     gr.Markdown(