File size: 886 Bytes
3e44479 4b57ca9 3e44479 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
---
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)
```
|