wlumsa / Khutbah_Summarizer

0 stars 0 forks source link

Khutbah Summarizer Task #1

Open UsamaMo opened 1 month ago

UsamaMo commented 1 month ago

Transcribe Youtube Video: https://pypi.org/project/youtube-transcript-api/

UsamaMo commented 1 month ago

https://huggingface.co/docs/transformers/model_doc/bart

image

UsamaMo commented 1 month ago

from transformers import AutoTokenizer, FlaxBartForConditionalGeneration

model = FlaxBartForConditionalGeneration.from_pretrained("facebook/bart-large-cnn") tokenizer = AutoTokenizer.from_pretrained("facebook/bart-large-cnn")

ARTICLE_TO_SUMMARIZE = "My friends are cool but they eat too many carbs." inputs = tokenizer([ARTICLE_TO_SUMMARIZE], max_length=1024, return_tensors="np")

Generate Summary

summary_ids = model.generate(inputs["input_ids"]).sequences print(tokenizer.batch_decode(summary_ids, skip_special_tokens=True, clean_up_tokenization_spaces=False))