Ryan Chesler commited on
Commit
ec58c6f
·
1 Parent(s): 554f1b8

updated demo and readme for package install

Browse files
Files changed (2) hide show
  1. Demo.ipynb +2 -2
  2. README.md +8 -3
Demo.ipynb CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:777844f15f935804cd331020a2338c7db3670a96b0dc6d3430b2602cde666e68
3
- size 989346
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4c6af3fabb5d255609ad3954c451015dbf565d3855ae54440f99429ba6978f55
3
+ size 878101
README.md CHANGED
@@ -134,7 +134,12 @@ git clone https://huggingface.co/nvidia/nemotron-graphic-elements-v1
134
  ```
135
  git clone [email protected]:nvidia/nemotron-graphic-elements-v1
136
  ```
137
-
 
 
 
 
 
138
  2. Run the model using the following code:
139
 
140
  ```
@@ -143,8 +148,8 @@ import numpy as np
143
  import matplotlib.pyplot as plt
144
  from PIL import Image
145
 
146
- from model import define_model
147
- from utils import plot_sample, postprocess_preds_graphic_element, reformat_for_plotting
148
 
149
  # Load image
150
  path = "./example.png"
 
134
  ```
135
  git clone [email protected]:nvidia/nemotron-graphic-elements-v1
136
  ```
137
+ Optional:
138
+ This can be installed as a package using pip
139
+ ```
140
+ cd nemotron-graphic-elements-v3
141
+ pip install -e .
142
+ ```
143
  2. Run the model using the following code:
144
 
145
  ```
 
148
  import matplotlib.pyplot as plt
149
  from PIL import Image
150
 
151
+ from nemotron_graphic_elements_v1.model import define_model
152
+ from nemotron_graphic_elements_v1.utils import plot_sample, postprocess_preds_graphic_element, reformat_for_plotting
153
 
154
  # Load image
155
  path = "./example.png"