fix readme
Browse files
README.md
CHANGED
@@ -10,12 +10,12 @@ you can read this [blogpost](https://huggingface.co/blog/not-lain/custom-archite
|
|
10 |
you can the model via the command
|
11 |
```python
|
12 |
from transformers import AutoModelForImageClassification
|
13 |
-
model = AutoModelForImageClassification.from_pretrained("not-lain/
|
14 |
```
|
15 |
or you can use the pipeline
|
16 |
```python
|
17 |
from transformers import pipeline
|
18 |
-
pipe = pipeline(model="not-lain/
|
19 |
pipe(
|
20 |
"url",
|
21 |
download=True, # will call the download_img method
|
|
|
10 |
you can the model via the command
|
11 |
```python
|
12 |
from transformers import AutoModelForImageClassification
|
13 |
+
model = AutoModelForImageClassification.from_pretrained("not-lain/mycustomarchitecture", trust_remote_code=True)
|
14 |
```
|
15 |
or you can use the pipeline
|
16 |
```python
|
17 |
from transformers import pipeline
|
18 |
+
pipe = pipeline(model="not-lain/mycustomarchitecture", trust_remote_code=True)
|
19 |
pipe(
|
20 |
"url",
|
21 |
download=True, # will call the download_img method
|