Spaces:

Stable-X
/

ReconViaGen

Running on Zero

Stable-X commited on Sep 3

Commit

11ec745

verified ·

1 Parent(s): 0bc6644

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -380,6 +380,7 @@ if __name__ == "__main__":
     pipeline = TrellisVGGTTo3DPipeline.from_pretrained("Stable-X/trellis-vggt-v0-1")
     pipeline.cuda()
     pipeline.VGGT_model.cuda()
     try:
         pipeline.preprocess_image(Image.fromarray(np.zeros((512, 512, 3), dtype=np.uint8)))    # Preload rembg
     except:

     pipeline = TrellisVGGTTo3DPipeline.from_pretrained("Stable-X/trellis-vggt-v0-1")
     pipeline.cuda()
     pipeline.VGGT_model.cuda()
+    pipeline.birefnet_model.cuda()
     try:
         pipeline.preprocess_image(Image.fromarray(np.zeros((512, 512, 3), dtype=np.uint8)))    # Preload rembg
     except: