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
|
@@ -3,5 +3,5 @@ library_name: transformers
|
|
| 3 |
pipeline_tag: summarization
|
| 4 |
---
|
| 5 |
# Trained Text Summerizer BART:
|
| 6 |
-
- This model is fine-tuned version of the
|
| 7 |
-
-
|
|
|
|
| 3 |
pipeline_tag: summarization
|
| 4 |
---
|
| 5 |
# Trained Text Summerizer BART:
|
| 6 |
+
- This model is fine-tuned version of the **facebook/bart-large-cnn** model.
|
| 7 |
+
- **TrainedTextSummerizerBART** is a model trained on **samsum** dataset from Hugging Face for summarization tasks.
|