Save lfs.

.gitignore

@@ -1,8 +1,5 @@
-
 render.obj
-
 .cog/
-
 __pycache__/
-
 output.obj
+output.bin

transformer_directory/model.safetensors → output.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2782fe5aab4625a1b595b311ceac27cd6198370f935998a8707dd81d6706fc12
-size 926663752
+oid sha256:ba39696e8eac4236b332aba9ecc4232831ef5ef12ad44b83d0e894766b9dd555
+size 25986

predict.py

@@ -2,11 +2,12 @@ from cog import BasePredictor, Input, Path
 import tempfile
 import time
 from meshgpt_pytorch import MeshTransformer, mesh_render
+from os.path import dirname
 
 class Predictor(BasePredictor):
     def setup(self):
         """Load the model into memory to make running multiple predictions efficient"""
-        self.transformer = MeshTransformer.from_pretrained("./transformer_directory")
+        self.transformer = MeshTransformer.from_pretrained("MarcusLoren/MeshGPT-preview")
 
     def predict(
         self,
@@ -51,10 +52,3 @@ class Predictor(BasePredictor):
         file_path = Path(tempfile.mkdtemp()) / file_name
         mesh_render.save_rendering(str(file_path), output)
         return file_path
-
-
-if __name__ == "__main__":
-    model_name = "MarcusLoren/MeshGPT-preview"
-
-    model = MeshTransformer.from_pretrained("MarcusLoren/MeshGPT-preview")
-    model.save_pretrained("./transformer_directory")