Update README.md
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/29004", 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/29004", trust_remote_code=True)
|
| 19 |
pipe(
|
| 20 |
"url",
|
| 21 |
download=True, # will call the download_img method
|