Instructions to use pratiknichite/TrainedTextSummerizerBART with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use pratiknichite/TrainedTextSummerizerBART with Transformers:
# Use a pipeline as a high-level helper # Warning: Pipeline type "summarization" 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("summarization", model="pratiknichite/TrainedTextSummerizerBART")# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForSeq2SeqLM tokenizer = AutoTokenizer.from_pretrained("pratiknichite/TrainedTextSummerizerBART") model = AutoModelForSeq2SeqLM.from_pretrained("pratiknichite/TrainedTextSummerizerBART", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -2,5 +2,6 @@
|
|
| 2 |
library_name: transformers
|
| 3 |
pipeline_tag: summarization
|
| 4 |
---
|
| 5 |
-
|
| 6 |
-
|
|
|
|
|
|
| 2 |
library_name: transformers
|
| 3 |
pipeline_tag: summarization
|
| 4 |
---
|
| 5 |
+
# Trained Facebook BART:
|
| 6 |
+
- The model is originally published by Facebook.
|
| 7 |
+
- This model has been further trained on 'samsum' dataset from Hugging Face for summarization tasks.
|