Instructions to use CreatorPhan/ViSummary with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use CreatorPhan/ViSummary 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="CreatorPhan/ViSummary")# Load model directly from transformers import AutoTokenizer, AutoModelForSeq2SeqLM tokenizer = AutoTokenizer.from_pretrained("CreatorPhan/ViSummary") model = AutoModelForSeq2SeqLM.from_pretrained("CreatorPhan/ViSummary") - Notebooks
- Google Colab
- Kaggle
Commit ·
7b8ee67
1
Parent(s): e4ae303
Update README.md
Browse files
README.md
CHANGED
|
@@ -28,5 +28,5 @@ tokens = tokenizer(f"Tóm tắt văn bản sau: {context}", return_tensors='pt')
|
|
| 28 |
output = model.generate(tokens.to(device), max_new_tokens=170)[0]
|
| 29 |
predict = tokenizer.decode(output, skip_special_tokens=True)
|
| 30 |
print(len(predict.split()))
|
| 31 |
-
predict
|
| 32 |
```
|
|
|
|
| 28 |
output = model.generate(tokens.to(device), max_new_tokens=170)[0]
|
| 29 |
predict = tokenizer.decode(output, skip_special_tokens=True)
|
| 30 |
print(len(predict.split()))
|
| 31 |
+
print(predict)
|
| 32 |
```
|