Update README.md
Browse files
README.md
CHANGED
|
@@ -61,8 +61,8 @@ from transformers import T5ForConditionalGeneration, T5Tokenizer
|
|
| 61 |
import torch
|
| 62 |
|
| 63 |
# Load model and tokenizer
|
| 64 |
-
model = T5ForConditionalGeneration.from_pretrained("
|
| 65 |
-
tokenizer = T5Tokenizer.from_pretrained("
|
| 66 |
|
| 67 |
# Generate summary
|
| 68 |
def generate_summary(text: str, max_length: int = 256) -> str:
|
|
|
|
| 61 |
import torch
|
| 62 |
|
| 63 |
# Load model and tokenizer
|
| 64 |
+
model = T5ForConditionalGeneration.from_pretrained("openchs/sum-flan-t5-base-synthetic-v1")
|
| 65 |
+
tokenizer = T5Tokenizer.from_pretrained("openchs/sum-flan-t5-base-synthetic-v1")
|
| 66 |
|
| 67 |
# Generate summary
|
| 68 |
def generate_summary(text: str, max_length: int = 256) -> str:
|