Spaces:

gpue
/

foundationpose

Sleeping

Georg commited on 8 days ago

Commit

9526ed3

1 Parent(s): cca7bfe

Prepare job build context

Files changed (1) hide show

app.py CHANGED Viewed

@@ -115,8 +115,7 @@ class FoundationPoseInference:
                 from estimator import FoundationPoseEstimator
                 self.model = FoundationPoseEstimator(
-                    device=str(self.device),
-                    weights_dir="weights"
                 )
                 if getattr(self.model, "available", True):
                     logger.info("✓ Real FoundationPose model initialized successfully")

                 from estimator import FoundationPoseEstimator
                 self.model = FoundationPoseEstimator(
+                    device=str(self.device)
                 )
                 if getattr(self.model, "available", True):
                     logger.info("✓ Real FoundationPose model initialized successfully")