|
--- |
|
tags: |
|
- custom_code |
|
--- |
|
# how to create custom architectures |
|
you can read this [blogpost](https://huggingface.co/blog/not-lain/custom-architectures-with-huggingface) to find out more 📖 |
|
|
|
# How to use |
|
you can the model via the command |
|
```python |
|
from transformers import AutoModelForImageClassification |
|
model = AutoModelForImageClassification.from_pretrained("not-lain/MyRepo", trust_remote_code=True) |
|
``` |
|
or you can use the pipeline |
|
```python |
|
from transformers import pipeline |
|
pipe = pipeline(model="not-lain/MyRepo", trust_remote_code=True) |
|
pipe( |
|
"url", |
|
download=True, # will call the download_img method |
|
clean_output=False # will be passed as postprocess_kwargs |
|
) |
|
``` |
|
# parameters |
|
the pipeline supports the following parameters : |
|
* download |
|
* clean_output |
|
|
|
you can also use the following method to download images from the web |
|
```python |
|
pipe.download_img(url) |
|
``` |
|
|
|
|
