add pipeline
#1
by not-lain - opened
this pr will add pipeline to your repo to test it out before merging you can try the following code :
pip install loadimg
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("image-to-image", model="p1atdev/MangaLineExtraction-hf",
revision="refs/pr/1",trust_remote_code=True)
btw can you tag me in more AI models that have custom architectures too ? (yours or any other person)
i would also like to share with you something that the team in huggingface have been working on which is the PyTorchModelHubMixin which will allow you to add push_to_hub from_pretrained save_pretrained to any custom model
I see. I'll mention you if I see any custom code.
Nice work👍
p1atdev changed pull request status to merged