Update summarizer.py
Browse files- summarizer.py +4 -2
summarizer.py
CHANGED
|
@@ -1,7 +1,9 @@
|
|
|
|
|
| 1 |
from transformers import BartTokenizer, BartForConditionalGeneration
|
| 2 |
|
| 3 |
-
|
| 4 |
-
|
|
|
|
| 5 |
|
| 6 |
def generate_summary(text: str) -> str:
|
| 7 |
inputs = tokenizer([text], max_length=1024, truncation=True, return_tensors="pt")
|
|
|
|
| 1 |
+
# summarizer.py
|
| 2 |
from transformers import BartTokenizer, BartForConditionalGeneration
|
| 3 |
|
| 4 |
+
# Жаңа fine-tuned модель жолы
|
| 5 |
+
tokenizer = BartTokenizer.from_pretrained("./bart-finetuned-arxiv")
|
| 6 |
+
model = BartForConditionalGeneration.from_pretrained("./bart-finetuned-arxiv")
|
| 7 |
|
| 8 |
def generate_summary(text: str) -> str:
|
| 9 |
inputs = tokenizer([text], max_length=1024, truncation=True, return_tensors="pt")
|