Instructions to use google/pix2struct-textcaps-base with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use google/pix2struct-textcaps-base with Transformers:
# Use a pipeline as a high-level helper # Warning: Pipeline type "image-to-text" is no longer supported in transformers v5. # You must load the model directly (see below) or downgrade to v4.x with: # 'pip install "transformers<5.0.0' from transformers import pipeline pipe = pipeline("image-to-text", model="google/pix2struct-textcaps-base")# Load model directly from transformers import AutoProcessor, AutoModelForImageTextToText processor = AutoProcessor.from_pretrained("google/pix2struct-textcaps-base") model = AutoModelForImageTextToText.from_pretrained("google/pix2struct-textcaps-base") - Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -59,7 +59,7 @@ Once saved, you can push your converted model with the following snippet:
|
|
| 59 |
from transformers import Pix2StructForConditionalGeneration, Pix2StructProcessor
|
| 60 |
|
| 61 |
model = Pix2StructForConditionalGeneration.from_pretrained(PATH_TO_SAVE)
|
| 62 |
-
processor = Pix2StructProcessor.from_pretrained(
|
| 63 |
|
| 64 |
model.push_to_hub("USERNAME/MODEL_NAME")
|
| 65 |
processor.push_to_hub("USERNAME/MODEL_NAME")
|
|
|
|
| 59 |
from transformers import Pix2StructForConditionalGeneration, Pix2StructProcessor
|
| 60 |
|
| 61 |
model = Pix2StructForConditionalGeneration.from_pretrained(PATH_TO_SAVE)
|
| 62 |
+
processor = Pix2StructProcessor.from_pretrained(PATH_TO_SAVE)
|
| 63 |
|
| 64 |
model.push_to_hub("USERNAME/MODEL_NAME")
|
| 65 |
processor.push_to_hub("USERNAME/MODEL_NAME")
|