Spaces:

jarif
/

Summarization

Running

jarif commited on Feb 26

Commit

0f97217

•

1 Parent(s): 89a4135

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ from transformers import BartForConditionalGeneration
 # Load the pre-trained model and tokenizer
 pretrained_model_name = "facebook/bart-large-cnn"
 hf_tokenizer = T5Tokenizer.from_pretrained(pretrained_model_name)
-learn = load_learner('article_highlights_part3.pkl')
 def summarize(article):
   # Preprocess the input text

 # Load the pre-trained model and tokenizer
 pretrained_model_name = "facebook/bart-large-cnn"
 hf_tokenizer = T5Tokenizer.from_pretrained(pretrained_model_name)
+learn = load_learner('article_highlights.pkl')
 def summarize(article):
   # Preprocess the input text