student2222333051 commited on
Commit
22143bf
·
verified ·
1 Parent(s): fbd3dd8

Update summarizer.py

Browse files
Files changed (1) hide show
  1. summarizer.py +4 -2
summarizer.py CHANGED
@@ -1,7 +1,9 @@
 
1
  from transformers import BartTokenizer, BartForConditionalGeneration
2
 
3
- tokenizer = BartTokenizer.from_pretrained("facebook/bart-large-cnn")
4
- model = BartForConditionalGeneration.from_pretrained("facebook/bart-large-cnn")
 
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")