BinLiunls commited on
Commit
45d6413
·
1 Parent(s): 5b420cf

update readme

Browse files

Signed-off-by: binliu <binliu@nvidia.com>

Files changed (1) hide show
  1. README.md +1 -2
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.