update readme
Browse filesSigned-off-by: binliu <binliu@nvidia.com>
README.md
CHANGED
@@ -24,7 +24,6 @@ with tempfile.TemporaryDirectory() as tmp_dir:
|
|
24 |
pipeline_helper = HuggingFacePipelineHelper("vista3d")
|
25 |
pipeline = pipeline_helper.init_pipeline(
|
26 |
os.path.join(FILE_PATH, "vista3d_pretrained_model"),
|
27 |
-
output_dir=output_dir,
|
28 |
device=torch.device("cuda:0"),
|
29 |
)
|
30 |
inputs = [
|
@@ -37,7 +36,7 @@ with tempfile.TemporaryDirectory() as tmp_dir:
|
|
37 |
"label_prompt": [3],
|
38 |
},
|
39 |
]
|
40 |
-
pipeline(inputs)
|
41 |
|
42 |
```
|
43 |
The inputs defines the image to segment and the prompt for segmentation.
|
|
|
24 |
pipeline_helper = HuggingFacePipelineHelper("vista3d")
|
25 |
pipeline = pipeline_helper.init_pipeline(
|
26 |
os.path.join(FILE_PATH, "vista3d_pretrained_model"),
|
|
|
27 |
device=torch.device("cuda:0"),
|
28 |
)
|
29 |
inputs = [
|
|
|
36 |
"label_prompt": [3],
|
37 |
},
|
38 |
]
|
39 |
+
pipeline(inputs, output_dir=output_dir)
|
40 |
|
41 |
```
|
42 |
The inputs defines the image to segment and the prompt for segmentation.
|