README.md

metadata

tags:
  - custom_code

how to create custom architectures

you can read this blogpost to find out more 📖

How to use

you can the model via the command

from transformers import AutoModelForImageClassification
model = AutoModelForImageClassification.from_pretrained("not-lain/mycustomarchitecture", trust_remote_code=True)

or you can use the pipeline

from transformers import pipeline
pipe = pipeline(model="not-lain/mycustomarchitecture", 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

pipe.download_img(url)