Fix the model path

Image3DProcessor.py

@@ -34,7 +34,7 @@ class Image3DProcessor:
         self.model_cfg = OmegaConf.load(model_cfg_path)
         
         # Load pre-trained model weights
-        model_path = hf_hub_download(repo_id=model_repo_id, filename=model_filename)
+        model_path = model_filename
         self.model = GaussianSplatPredictor(self.model_cfg)
         ckpt_loaded = torch.load(model_path, map_location=self.device)
         self.model.load_state_dict(ckpt_loaded["model_state_dict"])

app.py

@@ -10,9 +10,9 @@ from io import BytesIO
 import numpy as np
 
 class VideoGenerator:
-    def __init__(self, model_cfg_path, model_repo_id, model_filename):
+    def __init__(self, model_cfg_path, model_filename):
         # Initialize the Image3DProcessor
-        self.processor = Image3DProcessor(model_cfg_path, model_repo_id, model_filename)
+        self.processor = Image3DProcessor(model_cfg_path, model_filename)
 
     def generate_3d_video(self, image):
         # Ensure the image is a PIL Image object
@@ -33,8 +33,7 @@ class GradioApp:
         # Initialize VideoGenerator with required paths and details
         self.video_generator = VideoGenerator(
             model_cfg_path="/home/user/app/splatter-image/gradio_config.yaml",
-            model_repo_id="szymanowiczs/splatter-image-multi-category-v1",
-            model_filename="model_latest.pth"
+            model_filename="/home/user/app/fused_text_to_3D.pth"
         )
 
     def launch(self):