Spaces:
Running
Running
File size: 633 Bytes
ff51eba 2d8e2b5 ff51eba 2d8e2b5 ff51eba 2d8e2b5 d82eb66 2d8e2b5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("Chillyblast/Bart_Summarization")
model = AutoModelForSeq2SeqLM.from_pretrained("Chillyblast/Bart_Summarization")
from transformers import pipeline
# Create a pipeline for text summarization
summarizer = pipeline("summarization", model=model, tokenizer=tokenizer)
# Example input for inference
dialogue = input(str("Enter the input:"))
# Perform inference
summary = summarizer(dialogue, max_length=500, min_length=300, do_sample=False)
# Print the summary
print("Summary:", summary[0]['summary_text'])
|