Update summarizer.py

summarizer.py

@@ -1,9 +1,11 @@
 # summarizer.py
 from transformers import BartTokenizer, BartForConditionalGeneration
 
-# Жаңа fine-tuned модель жолы
-tokenizer = BartTokenizer.from_pretrained("./bart-finetuned-arxiv")
-model = BartForConditionalGeneration.from_pretrained("./bart-finetuned-arxiv")
+# HuggingFace Hub-тағы fine-tuned модель
+MODEL_HUB_NAME = "username/bart-finetuned-arxiv"  # өз username қой
+
+tokenizer = BartTokenizer.from_pretrained(MODEL_HUB_NAME)
+model = BartForConditionalGeneration.from_pretrained(MODEL_HUB_NAME)
 
 def generate_summary(text: str) -> str:
     inputs = tokenizer([text], max_length=1024, truncation=True, return_tensors="pt")
@@ -17,3 +19,4 @@ def generate_summary(text: str) -> str:
     )
 
     return tokenizer.decode(summary_ids[0], skip_special_tokens=True)
+