| --- |
| 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) |
| ``` |
| |
| |
