Download config from HF

app.py

@@ -29,16 +29,12 @@ def main():
     else:
         device = "cpu"
 
-    model_cfg = OmegaConf.load(
-        os.path.join(
-            os.path.dirname(os.path.abspath(__file__)), 
-                    "config.yaml"
-                    ))
-
-    model_path = hf_hub_download(repo_id="szymanowiczs/splatter-image-multi-category-v1", 
+    model_cfg_path = hf_hub_download(repo_id="szymanowiczs/splatter-image-v1", 
+                                 filename="config_objaverse.yaml")
+    model_path = hf_hub_download(repo_id="szymanowiczs/splatter-image-v1", 
                                  filename="model_latest.pth")
 
-    
+    model_cfg = OmegaConf.load(model_cfg_path)    
     model = GaussianSplatPredictor(model_cfg)
 
     ckpt_loaded = torch.load(model_path, map_location="cpu")