Ryan Chesler
commited on
Commit
·
ec58c6f
1
Parent(s):
554f1b8
updated demo and readme for package install
Browse files- Demo.ipynb +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:
|
| 3 |
-
size
|
|
|
|
| 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"
|